ML Engineer Internship, Enhance Hardware Efficiency - US Remote

Here at Hugging Face, we're on a mission to advance quality Machine Learning and make it more accessible. In the process, we contribute to the improvement of technology for the better.

We've created the fastest-growing, open-source library of pre-trained models in the world. With more than 500K+ models and 250K+ stars on GitHub, over 15,000 companies are using Hugging Face technology in production, including leading AI organizations like Google, Elastic, Salesforce, Algolia, and Grammarly.

About the Role

This internship aims to enhance the performance of transformer-based AI models by exploring how to utilize model compilation and graph rewriting to maximize hardware efficiency and performance. A comprehensive examination of various AI model optimization techniques, such as device placement, subgraph fusion, dispatching, and others, will be undertaken. These will be used to effectively facilitate performance improvements across various AI model topologies and tasks.

About You

If you're a fan of open-source, have a passion for making complex technology more accessible to engineers and artists, and want 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 check all the boxes above, we still encourage you to apply! We're establishing a diverse team whose skills, experiences, and backgrounds complement each other. We're open to considering where you might have the most significant impact.

More about Hugging Face

We're actively striving to build a culture that values diversity, equity, and inclusivity. We're consciously creating a workplace where everyone feels respected and supported—irrespective of who you are or where you come from. We believe that this is fundamental in building a great company and community. Hugging Face is an equal opportunity employer and we do not discriminate based on race, religion, color, nationality, gender, sexual orientation, age, marital status, veteran status, or disability status.

We prioritize growth. You'll get to work with some of the most intelligent people in our industry. We're an organization that leans towards impact and we constantly challenge ourselves to keep growing. We offer all employees reimbursement for relevant conferences, training, and education.

We care about your wellbeing. We provide flexible working hours and remote options. We support our employees wherever they're located. While we have office spaces around the world, particularly in the US, Canada, and Europe, we're highly distributed and all remote employees have the chance to visit our offices. If needed, we'll also equip your workstation to ensure your success.

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