ML Research Engineer III

We are seeking a strong Machine Learning Engineer to join our Fundamental AI Research team, an organization focused on making research breakthroughs in AI Responsibilities include developing deep learning libraries that support large-scale distributed training, open sourcing high quality code and reproducible results for the community, and bringing the latest research to client's products for connecting billions of users. The chosen candidate will work with a diverse and highly interdisciplinary team of scientists, engineers, and cross-functional partners, and will have access to cutting edge technology, resources, and research facilities.

Job 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

Skills:

2-5 years of Python experience 1-3 years of Pytorch experience 1-2 years of Distributed ML Training (FSDP/DDP) experience 1-3 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

Education/Experience:

2+ years of experience with deep learning Experience developing machine learning algorithms in Python or C/C++ Experience with machine learning frameworks such as PyTorch 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. Degree in Computer Science, Computer Engineering or relevant technical field