Open-Source Machine Learning Engineer

We're on a journey to democratize good AI. As an Open-Source Machine Learning Engineer, you'll work to improve the open-source machine learning ecosystem. You'll mainly work on existing open-source libraries such as Transformers, Datasets, Pytorch and vLLM, and you'll interact with users and contributors across the broad open-source ML ecosystem.

You'll help foster one of the most active machine learning communities, helping users contribute to and use the tools you build. You'll work with researchers, ML practitioners, and data scientists every day through GitHub, our forums, and Slack.

You have a public track record of open-source work, and you enjoy collaborating with a community out in the open on GitHub. You love open source, you're passionate about making complex technology more accessible, and you want to contribute to one of the fastest-growing ML ecosystems.

Responsibilities

• Work on existing open-source libraries such as Transformers, Datasets, Pytorch and vLLM
• Interact with users and contributors across the broad open-source ML ecosystem
• Help foster one of the most active machine learning communities
• Work with researchers, ML practitioners, and data scientists every day through GitHub, our forums, and Slack

Requirements

• Strong Python skills, with experience writing clean, well-tested, maintainable library code
• Deep hands-on experience with a modern deep-learning framework, especially PyTorch (JAX or TensorFlow a plus)
• Practical experience with the Hugging Face open-source stack (Transformers, Datasets, Accelerate) or comparable ML libraries
• A public track record of open-source contributions, for example merged pull requests to ML or data libraries, that we can review on GitHub
• Solid understanding of modern machine learning and deep learning, including transformer architectures
• Experience collaborating with a technical community in the open (GitHub issues and reviews, forums, Slack or Discord)

Nice to Have

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