Senior Applied Scientist - Sovereign AI at NVIDIA

You're a highly autonomous Senior Applied Scientist with a Masters or PhD in Computer Science, Machine Learning, or a related field, possessing 8+ years of technical experience with at least 4 years dedicated to the LLM/Deep Learning lifecycle. **What makes it worth a look...** This full-time, on-site role at NVIDIA in Pune, India, offers the chance to work on core Sovereign AI engineering efforts. You'll have the opportunity to design architecture ablations, write high-performance code, and upstream your learnings into NVIDIA's core libraries. **You might be a good fit if you...** * Have expert-level Python and PyTorch skills, with a focus on scalable, production-grade code. * Possess hands-on experience modifying large-scale frameworks like Megatron-LM, NeMo, TensorRT-LLM, vLLM, or Triton. * Have a deep understanding of Transformers, scaling laws, and training dynamics for LLMs. * Can lead end-to-end model training, evaluation, and efficiency optimization, including advanced compression techniques.