ML Engineer (Training Infra), Foundational Models
Sarvam · Bengaluru
CHECKED 18H AGO
Frontier labonsitemid-level
At a glance
Highlights
- Own the infrastructure behind Sarvam’s next family of foundational models.
- Make frontier training runs finish reliably, with impact in every model shipped.
- Build sovereign AI for India with real population-scale impact.
Ideal candidate
A systems-obsessed engineer who lives for reliability and performance at massive scale. You thrive on high ownership, move fast, and want your work to power sovereign AI for India.
Requirements
- BS or MS in Computer Science or closely related technical field or equivalent experience
- 3+ years building ML training infrastructure or large-scale distributed systems
- Hands-on experience training large models with distributed frameworks and on-call for pretraining runs
- Deep knowledge of GPU architecture, CUDA programming model and GPU profiling tools
- Strong PyTorch internals knowledge and comfort modifying low-level training code
- Meaningful open-source contributions in the training infrastructure ecosystem
Lab profile
About Sarvam
Building India's full-stack sovereign AI platform across research, models, infrastructure and applications.
What they build
- Model API
- On-device AI
- Enterprise AI agents
Known for
- Sovereign AI focus for India
- Full-stack across research to applications
- Production deployment at banks, insurers and government
- Indian-language interactions at scale
Key facts
- Backed by Lightspeed, Peak XV and Khosla Ventures