Machine Learning Engineer

Tech Giant is seeking a strong Senior Machine Learning Engineer to join our Fundamental AI Research (FAIR) team, an organization focused on making research breakthroughs in AI. The chosen candidate will develop deep learning libraries, contribute to open-source projects, and collaborate with a diverse team to bring the latest research to products connecting billions of users.

Responsibilities:

• Engineer, design, implement, and improve highly-scalable machine learning systems and tools for enabling research
• Apply knowledge of relevant research domains, along with expert coding skills, to platform and framework development projects
• Write clean and robust machine learning code
• Contribute to open-source projects

Requirements:

• Degree in Computer Science, Computer Engineering or relevant technical field
• 5+ years experience with deep learning
• Experience developing machine learning algorithms or machine learning infrastructure in Python or C/C++
• Experience with machine learning frameworks such as PyTorch and distributed optimization techniques such as FSDP/DDP
• Experience working with large datasets and data pipelines
• Solid understanding of algorithms, data structures, and software engineering best practices
• Demonstrated ability to work collaboratively in a fast-paced, team-oriented environment
• Excellent problem-solving and communication skills
• 5-10 years of Python experience
• 3-5 years of Pytorch experience
• 5+ years of Distributed ML Training (FSDP/DDP) experience
• 3-5 years of Dataset / Pytorch DataLoader experience
• Demonstrated software engineering experience via work experience, or widely used contributions in open source repositories (e.g. GitHub)
• Prior contributions to open-source AI/ML projects