Deep Learning Researcher

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 Skills required for this job: • C++ • Data pipelines • Deep learning • Delivered Duty Paid (DDP) • Distributed training • Fully Sharded Data Parallel (FSDP) • Machine learning • PyTorch • Python • Software engineering • TensorFlow