Research Software Engineer — Scalable RL & Distributed Training

Reflection AI; New York (NY); On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Top-tier compensation

Comprehensive medical, dental, vision insurance

Fully paid parental leave

Paid time off

Daily provided lunch and dinners

Job summary

A pioneering AI company in New York is seeking a skilled engineer to architect and optimize AI training infrastructure. The role involves working with distributed training systems and large-scale data pipelines. Ideal candidates have experience in performance optimization and implementing advanced techniques in machine learning. The company offers top-tier compensation, comprehensive health benefits, and opportunities for career growth in a collaborative environment.

Qualifications

• Strong software engineering skills with experience in distributed systems.
• Ability to implement research papers into practical applications.
• Experience in designing large-scale data pipelines and training systems.

Responsibilities

• Design and optimize large-scale training loops and data pipelines.
• Build tooling for monitoring and reproducing experiments.
• Diagnose bottlenecks across the training stack.

Skills

Distributed Training & Inference

Data Infrastructure

Performance optimization

Numerical stability

Tools

PyTorch

Kubernetes

Ray

Slurm

Job description

A pioneering AI company in New York is seeking a skilled engineer to architect and optimize AI training infrastructure. The role involves working with distributed training systems and large-scale data pipelines. Ideal candidates have experience in performance optimization and implementing advanced techniques in machine learning. The company offers top-tier compensation, comprehensive health benefits, and opportunities for career growth in a collaborative environment.