Staff Research Software Engineer - AI Training Infra

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

Benefits offered by this job

Top-tier compensation

Health & wellness benefits

Paid parental leave

Flexible paid time off

Daily meals and team events

Job summary

A leading AI development company located in New York is seeking a Software Engineer to develop groundbreaking infrastructure for AI model training. The role involves designing large-scale data pipelines and optimizing distributed systems. Candidates should have strong skills in distributed training or data infrastructure, with a focus on performance and reproducibility. The company offers top-tier compensation, comprehensive health benefits, and a supportive work culture.

Qualifications

• Strong software engineer with machine learning experience.
• Ability to implement research concepts without a PhD.
• Deep experience in distributed training or data infrastructure.
• Passion for performance, stability, and reproducibility.

Responsibilities

• Architect core training infrastructure for models.
• Design and optimize large-scale training loops.
• Implement state-of-the-art techniques for efficiency.
• Build tools for monitoring and reproducing experiments.
• Translate prototypes into production-grade systems.

Skills

Distributed Training & Inference

Data Infrastructure

GPU parallelism

Performance optimization

Tools

PyTorch

JAX

Kubernetes

Ray

Job description

A leading AI development company located in New York is seeking a Software Engineer to develop groundbreaking infrastructure for AI model training. The role involves designing large-scale data pipelines and optimizing distributed systems. Candidates should have strong skills in distributed training or data infrastructure, with a focus on performance and reproducibility. The company offers top-tier compensation, comprehensive health benefits, and a supportive work culture.