Open-Source Machine Learning Engineer
Build and improve open-source machine learning libraries while collaborating with researchers, practitioners, users, and contributors. The role requires strong Python, deep-learning framework experience, practical familiarity with the Hugging Face ecosystem, and a public record of open-source contributions.
About the job
Responsibilities
- Improve the open-source machine learning ecosystem.
- Work on open-source libraries such as Transformers, Datasets, PyTorch, and vLLM.
- Collaborate with users and contributors across the open-source ML ecosystem.
- Help foster the machine learning community and make tools accessible to users and contributors.
- Collaborate with researchers, ML practitioners, and data scientists through GitHub, forums, and Slack.
Requirements
- Strong Python skills and experience writing clean, well-tested, maintainable library code.
- Hands-on experience with a modern deep-learning framework, especially PyTorch; JAX or TensorFlow experience is a plus.
- Practical experience with the Hugging Face open-source stack, including Transformers, Datasets, and Accelerate, or comparable ML libraries.
- A public track record of open-source contributions, such as merged pull requests to machine learning or data libraries.
- Solid understanding of modern machine learning and deep learning, including transformer architectures.
- Experience collaborating with technical communities through 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.
- 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.
Compensation and Benefits
- Flexible working hours and remote options.
- Health, dental, and vision benefits for employees and their dependents.
- Parental leave and flexible paid time off.
- Reimbursement for relevant conferences, training, and education.
- Company equity for all employees.
- Access to office spaces in NYC and Paris, with opportunities for remote employees to visit.
- Workstation equipment support when needed.
Application Requirement
- Submit a cover letter explaining why you would like to work in open source at Hugging Face, including relevant skills, expertise, and topics of interest.
Skills
Python, PyTorch, JAX, TensorFlow, Transformers, Datasets, Accelerate, vLLM, Machine Learning, Deep Learning, Transformer Architectures, Distributed Training, Inference Optimization, Gpu Performance, GitHub
Similar jobs
ML Engineering jobs
Protege
Remote
Build the technical foundation for a new business vertical, creating reusable infrastructure and leading early customer engagements from scoping through delivery. The role requires 3+ years of engineering experience, strong Python and SQL skills, backend/data expertise, and comfort operating in ambiguity.
Beacon Biosignals
Boston, MADevelop and productionize machine- and deep-learning algorithms for biosignal and EEG data used in medical devices, clinical development, and diagnostics. The role requires 4+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated environments and production ML practices.
Adaption Labs
San Francisco, CA
Build production agent systems that plan, use tools, recover from failures, and improve over time. The role requires 5+ years of production ML or backend experience, LLM or agent deployment experience, and expertise in evaluation, tracing, observability, and agent architecture.
Adaption Labs
San Francisco, CA
Own inference-stack cost and performance by optimizing serving, caching, batching, quantization, decoding, routing, and GPU execution. The role requires 5+ years in ML systems, inference infrastructure, or performance engineering, plus strong Python and systems-language skills.
Cohere
New York, NYBuild and advance multilingual language models through scalable ML systems, research, and large-scale data processing. The role requires deep NLP expertise, strong Python and software engineering skills, and a PhD or equivalent experience, with opportunities to publish research and mentor teammates.