Open-Source Machine Learning Engineer - US Remote
Hugging Face5 months ago
Remote, United StatesMid Level
H1B sponsor
Responsibilities
- Improve and maintain open-source machine learning libraries such as Transformers, Datasets, PyTorch, and vLLM.
- Write clean, well-tested, and maintainable Python library code.
- Help users and contributors adopt and improve the tools through GitHub, forums, Slack, and other community channels.
- Collaborate with researchers, ML practitioners, and data scientists across the open-source ecosystem.
- Foster an active machine learning community and make complex technology more accessible.
Requirements
- Strong Python skills and experience writing clean, well-tested, maintainable library code.
- Deep hands-on experience with a modern deep-learning framework, especially PyTorch; JAX or TensorFlow is a plus.
- Practical experience with the Hugging Face open-source stack, including Transformers, Datasets, and Accelerate, or comparable ML libraries.
- A public, reviewable 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 with technical communities through GitHub issues and reviews, forums, Slack, or Discord.
- Fluent written English for asynchronous collaboration across a distributed global community.
- Preferred qualifications include open-source project maintenance, contributions to Transformers, Datasets, or Accelerate, distributed training, inference optimization, GPU or accelerator performance work, and model training or fine-tuning at scale.
- Applicants should provide a cover letter explaining their interest in open-source work at Hugging Face.
Benefits
- Fully remote work with flexible working hours.
- Health, dental, and vision benefits for employees and dependents.
- Parental leave and flexible paid time off.
- Reimbursement for relevant conferences, training, and education.
- Company equity as part of the compensation package.
- Opportunity for remote employees to visit the New York and Paris offices.
- Workstation equipment provided if needed.
Categories
About Hugging Face
Hugging Face builds an open platform and tools for machine learning developers to share, build, and deploy models, including the Hugging Face Hub, Transformers, and hosted inference services. Its business model combines open-source libraries with paid managed inference, enterprise hosting, and commercial partnerships. Founded in 2016 and privately held, the company serves a global developer community with millions of models, datasets, and apps hosted and reports over 11 million users.