Senior Software Engineer, Generative AI Research

What you'll do

• Design and build scalable infrastructure for training AI models like Cosmos.
• Develop high-throughput systems for efficient data processing and retrieval.
• Enhance system reliability and performance in distributed compute environments.
• Contribute to the long-term strategy for managing large-scale data and compute.
• Optimize massive data pipelines and support foundation model training workflows.

What we're looking for

• Strong engineering background in distributed systems and ML infrastructure with 6+ years of experience.
• Proficiency in Python and at least one systems language (C++, Go, Rust).
• Experience with orchestration systems, scheduling, and scalable storage/data pipelines.
• Ability to collaborate across teams and deliver robust solutions in complex environments.
• Hands-on experience building or optimizing infrastructure for large-scale model training.