Machine Learning Engineer Intern

Summary

Join Hugging Face's ML Optimization team as an intern and contribute to the democratization of machine learning. You will develop and refine cutting-edge solutions for various hardware platforms, creating user-friendly tools and guides for model conversion and deployment. Your work will involve optimizing hardware selection, conducting inference experiments, and sharing your findings with the Hugging Face community through various channels. This internship offers the opportunity to make a tangible impact on the AI landscape while collaborating with world-class experts and hardware providers. Hugging Face values diversity, equity, and inclusivity, and offers benefits such as flexible working hours, remote options, and reimbursement for relevant conferences and training.

Responsibilities

• Develop an online exporter tool: Create a user-friendly online tool to convert Hugging Face models for specific hardware platforms leveraging Optimum
• Bake the recipes: Author comprehensive guides to help users deploy Hugging Face models on various hardware platforms, including detailed instructions and best practices
• Design User Flow: Develop a seamless flow to guide users from traditional Hugging Face libraries (like Transformers and Diffusers) to alternative hardware backends. This includes integrating these solutions into the Hugging Face Hub and our partners' platforms
• Optimize Hardware Selection: Conduct inference experiments across different hardware backends to identify the strengths and weaknesses of each platform under various scenarios. Provide clear guidelines to help users select the best hardware for their specific tasks
• Advocate and Communicate Insights: Collaborate with the Hugging Face Advocacy team to share your findings and insights through various channels, including blog posts, tweets, leaderboards, Spaces, and YouTube videos. You will educate and inspire the community about the importance of hardware in AI

Benefits

• We provide all employees with reimbursement for relevant conferences, training, and education
• We offer flexible working hours and remote options