{"id":40228,"url":"https://alion.io/job/sarvam-ai-ml-engineer-training-infra-foundational-models","title":"ML Engineer (Training Infra), Foundational Models","company":{"id":229,"name":"Sarvam AI","domain":"sarvam.ai","url":"https://alion.io/company/sarvam-ai","size_band":"501-1000","is_staffing_agency":false,"employer_type":"direct","is_intermediary":false,"listed_via":null,"ats_vendor":"Ashby","truth_index":{"grade":"A","score":95,"open_postings":37,"ghost_share":0,"stale_share":0,"repost_share":0,"time_to_fill_p50_days":91,"computed_at":"2026-11-03T05:54:40Z"}},"role":"AI/ML","role_family":"AI/ML","seniority":"middle","employment_type":"full_time","work_mode":"on_site","remote_scope":null,"remote_scope_basis":null,"remote_working_hours":null,"hiring_geo_confidence":"structured","locations":["Bengaluru, India"],"countries":["IN"],"hiring_countries":[],"hiring_countries_total":0,"salary":null,"salary_estimate":{"min_usd":23000,"max_usd":56000,"period":"year","method":"role_seniority_country_remote_cell","sample_n":32},"experience_years_min":3,"visa_sponsorship":false,"relocation_package":false,"has_equity":false,"technologies":[{"name":"CUDA","optional":false},{"name":"DeepSpeed","optional":false},{"name":"FSDP","optional":false},{"name":"Megatron-LM","optional":false},{"name":"NCCL","optional":false},{"name":"NVIDIA NeMo","optional":false},{"name":"PyTorch","optional":false},{"name":"Triton","optional":false},{"name":"vLLM","optional":false},{"name":"Kubernetes","optional":true},{"name":"Quantization","optional":true},{"name":"SLURM","optional":true}],"status":"live","first_seen_at":"2026-05-21T07:28:11Z","employer_posted_date":"2026-10-07","last_verified_at":"2026-11-03T04:29:24Z","board_verified":true,"closed_at":null,"days_open":165,"trust":{"level":"ok","repost_count":1,"flags":[],"days_open":165},"description":"About Sarvam\nSarvam is building the bedrock of Sovereign AI for India. The company is developing India’s full-stack sovereign AI platform, building across research, models, infrastructure and applications with a singular focus on making AI genuinely work for India. Sarvam works with leading enterprises and public institutions and is backed by Lightspeed, Peak XV, and Khosla Ventures. Sarvam partners with India’s leading brands, including Tata Capital, SBI Life, CRED, IDFC, and LIC.\nAbout the Role\nYou will own the infrastructure that our next family of foundational models is trained on. This is a deep systems role: distributed training at scale, parallelism strategies, GPU kernel work, throughput and stability optimization, and the unglamorous reliability engineering that determines whether a frontier training run actually finishes.\nThe bar is high. A bad MFU number, a slow data loader, a switch flap - these are problems whose cost is measured in weeks and millions of dollars. You will be expected to find them before they happen, and to fix them fast when they do.\nWhat You’ll Do\nBuild, maintain, and continually push the limits of our distributed training stack across large GPU clusters.\n\nDesign and implement parallelism strategies - data, tensor, pipeline, sequence, expert - and reason about which combinations make sense for which architectures and which scales.\n\nProfile and optimize end-to-end training throughput: kernel performance, communication overlap, memory layout, checkpointing, data loading.\n\nWrite and tune custom GPU kernels (CUDA, Triton) where off-the-shelf is leaving performance on the table.\n\nOwn the reliability of long-running training jobs - fault tolerance, checkpoint integrity, deterministic restarts, automated detection of slow nodes and silent corruption.\n\nPartner closely with researchers to make sure architectural ideas can actually be trained efficiently, and with the data team to keep the pipeline from being the bottleneck.\n\nWhat We’re Looking For\nBS or MS in Computer Science or a closely related technical field (or equivalent demonstrated experience).\n\n3+ years of experience building ML training infrastructure or large-scale distributed systems. Exceptional early-career candidates with a strong systems background will be considered.\n\nHands-on experience training large models with distributed training frameworks - Megatron-LM, DeepSpeed, FSDP, NeMo, or equivalent. You should have been on-call for a real pretraining run.\n\nDeep working knowledge of GPU architecture, the CUDA programming model, and standard tools for profiling GPU workloads (Nsight, PyTorch profiler, etc.).\n\nStrong PyTorch internals knowledge. You're comfortable reading and modifying low-level training code, not just using high-level APIs.\n\nMeaningful open-source contributions in the training infrastructure ecosystem - Megatron, DeepSpeed, PyTorch, vLLM, Triton, NCCL, or similar.\n\nBonus Points\nCustom CUDA or Triton kernel development with measurable wins on real workloads.\n\nDirect experience training models at 10B+ parameters or on 1000+ GPU clusters.\n\nFamiliarity with cluster orchestration and job scheduling (Slurm, Kubernetes) at scale.\n\nExperience with mixed precision (BF16, FP8), quantization-aware training, or other numerical work that has shown up in production training runs.\n\nFirst-author papers or technical reports on training systems, scaling, or model efficiency.\n\nWhy this role?\nFrontier model training is one of the hardest systems problems in the world right now, and most of the people doing it well sit at a small handful of labs. You will be doing the same work, with the same autonomy, and your impact will be visible in every model Sarvam ships.\nWhy Sarvam?\nSarvam is a fast-moving, high talent-density team building full-stack AI for India, working on problems that push the frontiers of AI with real population-scale impact.\nWork alongside researchers, engineers, builders, and business leaders who move fast and hold each other to a very high bar\n\nHigh ownership and high impact, from day one\n\nEverything we do is AI-first, from the way we build and ship to the way we think about problems\n\nYou can work on problems that could change how an entire country learns, works, and communicates\n\nIf you want to work on problems at the frontier of AI in India, Sarvam is the place to be.","description_format":"text","description_chars":4347,"description_truncated":false,"requirements":{"experience_years_min":3,"management_years_min":null,"team_size_min":null,"manages_managers":false,"education":{"level":"bachelor","optional":false},"security_clearance":false,"languages":[]},"benefits":[],"hiring_locations":[],"hiring_excludes":[],"relocation_offered":false,"industries":["LLM & Generative AI","Natural Language Processing","Speech & Audio AI","Foundation Models"],"lifecycle":[{"event":"open","at":"2026-05-21T07:28:11Z"},{"event":"close","at":"2026-08-31T20:53:16Z"},{"event":"reopen","at":"2026-10-07T12:24:13Z"}],"visa":[],"liveness":{"score":23,"band":"cold","label":"Long shot","p_open":1,"p_active":0.647,"p_room":0.36,"age_days":165,"expected_fill_days":91,"reasons":["conf:1","velocity","win:tail","crowd:"],"computed_at":"2026-11-03T05:54:40Z"},"pay":null,"html_url":"https://alion.io/job/sarvam-ai-ml-engineer-training-infra-foundational-models","json_url":"https://alion.io/job/sarvam-ai-ml-engineer-training-infra-foundational-models.json","meta":{"generated_at":"2026-11-03T06:03:56Z","cache_seconds":300,"methodology":"https://alion.io/methodology","terms":"https://alion.io/terms","contact":"https://alion.io/contact","api":"https://alion.io/developers","about":"Alion is a live layer of people, companies and AI agents: who they are, whether they are real and active right now, what they do and how to work with them, readable by people and by agents and paid per call.","catalog":"https://alion.io/catalog.json","usage":{"tier":"crawler_verified","counted_by":"address","units_charged":1,"used_today":8225,"day_limit":null,"remaining_today":null,"minute_limit":300,"resets_at":"2026-11-04T00:00:00Z"}},"offers":[{"id":"company.slices","title":"One company in depth, by slice","status":"live","price":{"credits":0.02,"usd":0.002,"plus_per_slice":{"credits":0.05,"usd":0.005}},"unit":"per company, plus each slice with data","note":"the employer in depth","call":{"mcp_tool":"get_company","arguments":{"id":229},"rest":"https://alion.io/mcp/rest/get_company?id=229"},"human":"https://alion.io/catalog?offer=company.slices&for=job%2Fsarvam-ai-ml-engineer-training-infra-foundational-models"},{"id":"market.stats","title":"A market slice: pay, demand and time to fill","status":"live","price":{"credits":1,"usd":0.1},"unit":"per slice","note":"pay, demand and time to fill for this role and place","call":{"mcp_tool":"market_stats"},"human":"https://alion.io/catalog?offer=market.stats&for=job%2Fsarvam-ai-ml-engineer-training-infra-foundational-models"},{"id":"job.search","title":"Open jobs by role, technology, place, pay and visa","status":"live","price":{"credits":0.02,"usd":0.002},"unit":"per posting in a list","note":"similar open postings","call":{"mcp_tool":"search_jobs"},"human":"https://alion.io/catalog?offer=job.search&for=job%2Fsarvam-ai-ml-engineer-training-infra-foundational-models"},{"id":"company.verify","title":"Is this company real and active right now","status":"pilot","price":null,"unit":"per company","request":{"url":"https://alion.io/catalog/request","method":"POST","body":"{\"offer\": \"company.verify\", \"for\": \"job/sarvam-ai-ml-engineer-training-infra-foundational-models\", \"note\": \"what you need it for\"}"},"human":"https://alion.io/catalog?offer=company.verify&for=job%2Fsarvam-ai-ml-engineer-training-infra-foundational-models"}],"agent":{"mcp":"https://alion.io/mcp/jobs","mcp_open":"https://alion.io/mcp/open","next":[{"tool":"get_job","arguments":{"id":40228}},{"tool":"search_jobs","arguments":{"company":"sarvam.ai","detail":"compact"}},{"tool":"match_people","arguments":{"job_id":40228},"note":"people open to offers who fit this vacancy"}],"note":"The same data through MCP tools: https://alion.io/mcp/open needs no sign-in (compact searches, postings, people); https://alion.io/mcp with an Alion account has everything. Docs: https://alion.io/developers"}}