NVIDIA Logo

NVIDIA

Senior System Software Engineer - Local AI

Reposted One Month Ago
Be an Early Applicant
In-Office
Pune, Maharashtra
Junior
In-Office
Pune, Maharashtra
Junior
Build and optimize high-performance local AI inference software for NVIDIA RTX and DGX GPUs. Develop inference runtimes and execution stacks, optimize models and data pipelines using quantization, pruning, sparsity, and distillation, and perform system-level debugging and performance analysis. Collaborate with software, research, architecture, and product teams to improve scalability, stability, accuracy, and production readiness across language, vision, speech, and diffusion workloads.
The summary above was generated by AI

NVIDIA has continuously reinvented itself for more than two decades. The invention of the GPU in 1999 fueled the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. GPU-powered deep learning helped ignite the era of modern AI, establishing GPUs as the foundation of intelligent applications across productivity, gaming, and creative workflows, and reinforcing NVIDIA's position as a leading AI computing company. 

More recently, there is a growing focus on running AI models locally, closer to where data is generated. This approach reduces latency, enables real-time processing, and addresses privacy concerns by minimizing the need to send data to centralized servers. As technology continues to evolve, client-side AI will play an increasingly important role in shaping the digital landscape. The LocalAI team is seeking a Senior Systems Software Engineer to lead the architecture of efficient on-device AI software for RTX and DGX-class systems. The role focuses on delivering high-performance local inference with low latency, optimized memory utilization, robust infrastructure, and practical deployment on resource-constrained platforms 


What You'll Be Doing: 

  • Partner with NVIDIA's software, research, architecture, and product leadership, as well as external partners such as Microsoft, to align technical requirements and strategic priorities, fostering the AI ecosystem on RTX and DGX PCs.  
  • Build and optimize the local AI inference stack for RTX, RTX Pro, and DGX GPUs, with a focus on performance, stability, and scalability across diverse hardware architectures.  
  • Architect and develop modern inference runtimes and execution stacks using frameworks such as llama.cpp, vLLM, PyTorch, WinML, DXCGC, and TensorRT-RTX, supporting LLM, vision-language, TTS, ASR, and diffusion-based AI workloads.  
  • Perform end-to-end optimization of AI models, data pipelines, and inference runtimes to maximize performance on current and next-generation GPU architectures. Apply model optimization techniques, including quantization, pruning, sparsity, and distillation, to enable efficient deployment of large models on local and edge devices.  
  • Lead system-level debugging, performance tuning, and performance-accuracy trade-off analysis; develop infrastructure for performance and accuracy sweeps; analyse 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. 
  • Mentor other engineers and review architecture proposals spanning several groups, acting as a force multiplier for technical quality and consistency across the broader LocalAI organization. 

  

What we need to see: 

  • 5+ Years of experience with Bachelor’s, Master’s, or PhD in Computer Science, Software Engineering, Mathematics, or related field, or equivalent experience. 
  • Excellent C++ programming and debugging skills, with a strong foundation in data structures, algorithms, and machine learning.  
  • Proven experience architecting and optimizing AI inference pipelines and applications using ML/DL frameworks such as Llama.cpp, vLLM, PyTorch, Windows ML, DXCGC, and TensorRT. 
  • Deep understanding of inference backends and runtime internals, including scheduling, memory management, KV-cache behavior, graph execution, quantization, and hardware-aware optimization. 
  • Demonstrated ability to establish technical direction and influence outcomes across multiple teams and organizations without relying on formal management authority.  
  • Strong analytical and problem-solving skills, with the ability to manage multiple priorities effectively in a fast-paced environment. 
  • Excellent written and verbal communication skills, enabling effective collaboration across engineering teams and management.  

Ways to stand out from the crowd: 

  • Deep understanding of modern machine learning, deep neural network, and generative AI techniques, with exposure to AOT graph-compilation systems (TensorRT builder, MLIR-based compilers) 
  • Consistent track record of delivering end-to-end products in multinational companies with geographically distributed teams.  
  • Expert-level low-level systems and GPU kernel programming, CUDA, and development of high-performance systems. 
  • Significant contributions to open-source inference runtimes, model tooling, or performance infrastructure, as a maintainer or core contributor.  
  • Hands-on experience building applications using frameworks and APIs such as llama.cpp, PyTorch, TensorRT, Vulkan, DirectX, and vLLM. 

We're a top employer recognized for innovation, growth, and commitment to diversity as an equal opportunity workplace. We offer competitive salaries, a generous benefits package, and the opportunity to work alongside some of the industry's most talented and forward-thinking professionals. As our engineering teams continue to grow rapidly, we're looking for creative, self-driven engineers with a passion for technology to join us. 

NVIDIA Mumbai, Maharashtra, IND Office

No. 127, Andheri Kurla Road, Village, Chakala, Andheri East, Mumbai, Maharashtra, India, 400093

Similar Jobs

2 Days Ago
In-Office
Senior level
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Develop system software and infrastructure for deploying, benchmarking, and qualifying local AI workloads across Linux and Windows GPU systems. Build CI pipelines, model management services, databases, data pipelines, dashboards, telemetry, scheduling systems, and autonomous infrastructure agents. Collaborate with inference and application teams to improve reproducibility, diagnose system issues, detect regressions, and optimize performance.
Top Skills: CC#C++GitGit-Based CiGrafanaKibanaLinuxLlama.CppNvidia Dgx GpusNvidia Rtx GpusOllamaPythonPyTorchSQLVllmWindowsWinml
9 Days Ago
In-Office
Senior level
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Build and maintain system software and infrastructure for deploying, benchmarking, automating, and qualifying AI applications and models. Responsibilities include supporting inference backends, managing local model repositories, developing performance and correctness measurement systems, processing workload metrics, implementing diagnostics and observability, and maintaining automated build, integration, deployment, and qualification pipelines. The role also involves debugging system architectures, analyzing bottlenecks, and improving scalability across Linux and Windows environments.
Top Skills: CC#C++DockerGitGrafanaJavaJenkinsKibanaKubernetesLinuxLlama.CppNsight SystemsOllamaPerforcePythonPyTorchSQLTrt-RtxVllmWindowsWinml
4 Minutes Ago
Hybrid
Mumbai, Maharashtra, IND
Senior level
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Leads partner-focused presales for ServiceNow transformation solutions. Conducts discovery workshops, product demonstrations, solution architecture discussions, technical scoping, competitive positioning, and partner enablement. Serves as a trusted technical advisor to sales, system integrators, customers, and executives while supporting pipeline development, co-selling, and go-to-market activities. Provides product feedback and thought leadership through workshops, events, white papers, and industry forums.
Top Skills: Advanced AnalyticsAICloud PlatformsCRMMulti-TenancyServicenowWorkflow Automation

What you need to know about the Mumbai Tech Scene

From haggling for the best price at Chor Bazaar to the bustle of Crawford Market, the energy of Mumbai's traditional markets is a key part of the city's charm. And while these markets will always have their place, the city also boasts a thriving e-commerce scene, ranking among the largest in the region. Driven by online sales in everything from snacks to licensed sports merchandise to children's apparel, the local industry is worth billions, with companies actively recruiting to meet the demands of continued growth.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account