Deep Learning Researcher

🌍 Remote, USA 🎯 Full-time 🕐 Posted Recently

Job Description

Job Title: Senior Research Engineer / Senior Machine Learning Engineer
Location: Remote – United States
Duration: 12 Months (W2 Contract, with potential extension)

About the Role
We are seeking a Senior Research Engineer to join a highly collaborative research and engineering team focused on building scalable deep learning systems and distributed training infrastructure. This role involves developing highu2011quality machine learning libraries, enabling largeu2011scale model training, and translating cuttingu2011edge research into realu2011world products that operate at massive scale.

You will work closely with scientists, engineers, and crossu2011functional partners to design, implement, and optimize deep learning solutions using modern frameworks and distributed systems.

Mustu2011Have Technical Skills
5–10 years of professional Python experience
3–5 years of experience with distributed machine learning training (e.g., FSDP, DDP, or similar approaches)
3–5 years of handsu2011on PyTorch experience
3–5 years working with datasets, data pipelines, and PyTorch DataLoader

Niceu2011tou2011Have Skills
Active or past contributions to openu2011source ML/AI repositories
Strong engineering background with a focus on scalable systems
Experience collaborating with research teams to productionize ML models

Responsibilities
Design, develop, and maintain deep learning libraries supporting largeu2011scale distributed training
Implement and optimize distributed training strategies using techniques such as Data Parallelism and Fully Sharded training
Build and maintain robust data pipelines and dataset loading systems for largeu2011scale training
Translate research ideas into productionu2011ready ML systems
Write clean, efficient, and wellu2011documented code with a strong emphasis on reproducibility
Contribute to openu2011source projects and publish highu2011quality, reusable code when applicable
Collaborate closely with researchers, engineers, and product partners in a fastu2011paced environment
Debug and optimize performance across GPUu2011based training systems

Qualifications
Bachelor’s, Master’s, or PhD in Computer Science, Computer Engineering, or a related technical field
5+ years of handsu2011on experience in deep learning and machine learning engineering
Strong experience developing ML algorithms or infrastructure using Python and/or C/C++
Experience with PyTorch and distributed training approaches (e.g., DDP, FSDP or equivalent)
Experience working with large datasets, data preprocessing, and data loading pipelines
Solid understanding of algorithms, data structures, and software engineering best practices
Proven ability to work effectively in a collaborative, teamu2011oriented environment
Strong problemu2011solving and communication skills

Preferred Qualifications
Demonstrated software engineering experience through professional work or widely used openu2011source contributions
Prior contributions to openu2011source AI/ML projects
Experience training large transformeru2011based or deep neural network models
Familiarity with performance optimization, memory efficiency, and scalable training systems

Location: Remote - United States