Sarvam logo
Posted 3mo agoBengaluru

Performance Engineer, On-Device Inference

MiddleOn-site (Bengaluru)Salary undisclosed
Required Skills
PyTorch
Job Description

About Sarvam

Sarvam 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.

About the Role

Take Sarvam models from research-handoff state to production-ready artifacts on at least two of our target chipsets (Intel xPU, ARM xPU, Apple xPU, Nvidia / AMD GPUs). You'll own 1 - 2 (model, chipset) pairs end-to-end and partner with the consuming app team during integration.

What You’ll Do

  • Quantize, validate accuracy, benchmark, and document. Author the deployment workbook for each pair you own.

  • Embed part-time with consuming teams during integration; debug perf and accuracy issues alongside them.

  • Maintain and extend the team's benchmark harness.

Similar Openings in Other

View all in category