Open-Source Machine Learning Engineer
As an Open-Source Machine Learning Engineer, you will enhance the open-source machine learning ecosystem, focusing on libraries like Transformers and PyTorch. You will collaborate with the ML community, contributing to and supporting the tools you build.
About the job
About the Role
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. We'll brainstorm with you to put you in a position to do the work that interests you and that is impactful.
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.
What you'll need
- 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)
- Fluent written English for asynchronous collaboration across a distributed, global community
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
Requirements
Please provide a cover letter mentioning why you would like to work in open-source at Hugging Face. We encourage you to mention your skills, potential expertise, and topics on which you would like to work.
Skills
Python, PyTorch, JAX, TensorFlow, Transformers, Datasets, Accelerate, Machine Learning, Deep Learning, Transformer Architectures
Similar jobs
ML Engineering jobs
Thinking Machines Lab
San Francisco, CA
Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.
OpenAI
San Francisco, CA
Build research infrastructure and tooling that enables AI models to design silicon, including reinforcement learning environments, EDA integrations, evaluations, and experiment workflows. The role requires strong software engineering fundamentals and comfort working across research, tooling, and chip-design systems.
Rollstack
United StatesBuild production AI capabilities for automated slide and document generation, working across LLM applications, data analysis, and content generation. The role requires 3+ years in machine learning and NLP, advanced Python, and experience with LLM frameworks and production systems.
ClickUp
United States
Build and operate large-scale ranking and retrieval systems that power search relevance, including hybrid lexical/vector search, embeddings, query understanding, and permission-aware retrieval. Requires a bachelor's degree and 5+ years of ML engineering experience in ranking or information retrieval.
PathAI
Boston, MADevelop and deploy machine learning models for biomedical research and AI products, collaborating with scientific, engineering, and product teams. Requires an advanced quantitative degree, substantial ML experience, Python proficiency, and experience bringing models into production or research applications.