Senior System Software Engineer - Local AI

NVIDIA NVIDIA · Semiconductors · Pune, India

Senior Systems Software Engineer to build efficient on-device AI software for RTX and DGX-class systems, focusing on high-performance local inference, low latency, efficient memory use, infrastructure, and practical deployment on resource-constrained platforms.

What you'd actually do

  1. Building and optimizing local AI inference stack for RTX, RTX Pro and DGX GPUs, focusing on performance, stability, and scalability across various hardware architectures.
  2. Architecture and development of modern inference runtimes and execution stacks, covering frameworks like Llama.cpp, vLLM, PyTorch, WinML, DXCGC, and TensorRT-RTX across LLMs, vision-language, TTS, ASR, and diffusion AI workloads.
  3. Perform end-to-end optimization of AI models, data pipelines, and inference runtimes to enhance performance across current and next-generation GPU architectures.
  4. Apply model optimization techniques such as quantization, pruning, sparsity, and distillation to enable efficient deployment of large models on local and edge devices.
  5. Perform system-level debugging, performance optimization, and performance–accuracy trade-off analysis; develop infrastructure for performance and accuracy sweeps, analyze results to identify gaps, and drive fixes; and establish engineering guidelines to accelerate bring-up and ensure production readiness of new models and inference backends.

Skills

Required

  • 5+ years of experience with Bachelor's, Master's, or PhD in Computer Science, Software Engineering, Mathematics, or a related field, or equivalent experience
  • Excellent C++ programming and debugging skills
  • strong understanding of data structures, algorithms and machine learning
  • Proven experience working with AI inferencing pipelines and applications using ML/DL frameworks, such as Llama.cpp, vLLM, PyTorch, WinML, DXCGC and TensorRT
  • Deep interest in inference backends and runtime internals, including scheduling, memory management, KV-cache behavior, graph execution, quantization, and hardware-aware optimization
  • Strong analytical and problem-solving abilities
  • capability to multitask effectively in a dynamic environment
  • Outstanding written and oral communication skills

Nice to have

  • Understanding of modern techniques in Machine Learning, Deep Neural Networks, and Generative AI
  • relevant contributions to major open-source projects
  • Consistent track record of delivering end-to-end products with geographically distributed teams in multinational product companies
  • Proficiency in lower-level system/GPU programming, CUDA, and developing high-performance systems
  • Contributions to open-source inference runtimes, model tooling, or performance infrastructure
  • Hands-on experience building applications with frameworks and APIs like Llama.cpp, PyTorch, TensorRT, Vulkan, and DirectX, vLLM

What the JD emphasized

  • excellent C++ programming and debugging skills
  • proven experience working with AI inferencing pipelines and applications using ML/DL frameworks
  • deep interest in inference backends and runtime internals

Other signals

  • on-device AI software
  • high-performance local inference
  • low latency
  • efficient memory use
  • infrastructure and practical deployment on resource-constrained platforms