Open-Source Machine Learning Engineer - US Remote

What you'll do

• Develop and improve open-source machine-learning libraries, including Transformers, Datasets, PyTorch, and vLLM.
• Collaborate with researchers, ML practitioners, data scientists, users, and contributors through GitHub, forums, and Slack.
• Help foster the open-source community and make complex technology more accessible.

What they ask for

• Strong Python skills and experience writing clean, well-tested, maintainable library code.
• Deep hands-on experience with a modern deep-learning framework, especially PyTorch; practical experience with the Hugging Face open-source stack or comparable ML libraries.
• Public track record of open-source contributions, such as merged pull requests to ML or data libraries.
• Solid understanding of modern machine learning and deep learning, including transformer architectures; experience collaborating in technical communities and fluent written English.

Nice to have

• Experience maintaining an open-source project or prior contributions to Transformers, Datasets, Accelerate, or similar libraries.
• Familiarity with distributed training, inference optimization, GPU/accelerator performance, or training or fine-tuning models at scale.

Tools

Python, PyTorch, JAX, TensorFlow, Transformers, Datasets, Accelerate, vLLM

AI focus

Improve open-source machine-learning tools and make AI technology more accessible; work may involve model training, fine-tuning, distributed training, inference optimization, or GPU/accelerator performance.