Internship for Machine Learning Engineer at Brrr/Nanotron - Remote in the US

Here at Hugging Face, we're on a journey to advance superior Machine Learning and make it more accessible. Along this path, we contribute to the enhancement of technology for the greater good.

We have constructed the world's fastest-growing, open-source, library of pre-trained models. With over 500K+ models and 250K+ stars on GitHub, more than 15,000 companies are utilizing HF technology in production. This includes leading AI organizations such as Google, Elastic, Salesforce, Algolia, and Grammarly.

About the Role

In training its LLM, Hugging Face utilizes a specific 3D parallelism distributed training library internally (which we intend to make open-source at some point). You'll join the small team responsible for maintaining and enhancing this library. You should take pleasure in delving into CUDA kernels and performance optimization.

About You

If you're passionate about open-source, have a fervor for making complex technologies more accessible to engineers and artists and wish to contribute to one of the fastest-growing ML ecosystems, then we can't wait to see your application!

If you're interested in joining us but don't meet every requirement we still encourage your application! We're building a diverse team whose skills, experiences, and backgrounds complement one another. We're more than willing to consider where you might have the most significant impact.

More about Hugging Face

We are actively cultivating a company culture that values diversity, equity, and inclusiveness. We are intentionally shaping a workplace where everyone feels respected and supported — regardless of who you are or where you come from. We believe that this is foundational to building a great company and community. Hugging Face is an equal opportunity employer, and we do not discriminate based on race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

We value personal growth. You will work with some of the most intelligent people in our industry. We are an organization that favors action and is continuously challenging ourselves to grow. We offer all employees reimbursement for relevant conferences, training, and education.

We care about your welfare. We offer flexible working hours and remote working options. We support our employees wherever they may be. Although we have office spaces worldwide, particularly in the US, Canada, and Europe, we're very distributed and all remote workers can visit our offices. If necessary, we'll also set up your workstation to ensure your success.

We support the community. We believe significant scientific advancements result from collaboration across the field. Join a community that supports the ML/AI community.