Senior System Software Engineer - LocalAI

Posted 3 Days Ago
Be an Early Applicant
Pune, Maharashtra, IND
In-Office
Senior level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role
Develop and optimize on-device AI inference stacks and runtimes for RTX and DGX systems. Collaborate with cross-functional teams to design high-performance, memory-efficient inference pipelines, apply model optimizations (quantization, pruning, distillation), perform system-level debugging and performance tuning, and build infrastructure for performance/accuracy evaluation and production readiness.
Summary Generated by Built In

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 develop 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 teams 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. 

  • Design 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.  

  • Conduct 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.  

 

What we need to see: 

  • 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, with a strong foundation in data structures, algorithms, and machine learning. 

  • Proven experience developing 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 behaviour, graph execution, quantization, and hardware-aware optimization. 

  • 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: 

  • Understanding of modern machine learning, deep neural network, and generative AI techniques, with relevant contributions to major open-source projects. 

  • Consistent track record of delivering end-to-end products in multinational companies with geographically distributed teams. 

  • Proficiency in low-level system and GPU programming, CUDA, and the development of high-performance systems. 

  • Contributions to open-source inference runtimes, model tooling, or performance infrastructure. 

  • 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 a 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 technology 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. 

Skills Required

  • 5+ years of experience with a degree in Computer Science or related field, or equivalent experience
  • Excellent C++ programming and debugging skills, strong foundation in data structures and algorithms
  • Proven experience developing and optimizing AI inference pipelines and applications using frameworks such as llama.cpp, vLLM, PyTorch, Windows ML, DXCGC, and TensorRT
  • Deep understanding of inference backends and runtime internals (scheduling, memory management, KV-cache behavior, graph execution, quantization, hardware-aware optimization)
  • Strong analytical and problem-solving skills; ability to manage multiple priorities in a fast-paced environment
  • Excellent written and verbal communication skills for cross-team collaboration
  • Proficiency in low-level system and GPU programming and CUDA
  • Contributions to open-source inference runtimes, model tooling, or performance infrastructure
  • Hands-on experience with Vulkan, DirectX, and building applications using llama.cpp, PyTorch, TensorRT, vLLM
  • Track record delivering end-to-end products in multinational, distributed teams

NVIDIA Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about NVIDIA and has not been reviewed or approved by NVIDIA.

  • Equity Value & Accessibility Equity awards and a discounted ESPP are highlighted as core parts of total compensation, enabling employees to share in the company’s success. Stock-based compensation and the two-year lookback ESPP are consistently described as especially valuable.
  • Healthcare Strength Health coverage is portrayed as robust, with comprehensive medical, dental, and vision options alongside mental health support and on-site care resources. Employer HSA contributions and wellness perks reinforce the depth of the offering.
  • Retirement Support Retirement programs are depicted as strong, featuring a meaningful 401(k) match with Roth options and support for Mega Backdoor Roth contributions. These elements position long-term savings as a notable advantage of the total rewards package.

NVIDIA Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Jobs

ZS Logo ZS

Decision Analytics Associate (CMC)

Artificial Intelligence • Healthtech • Professional Services • Analytics • Consulting
Hybrid
2 Locations
15000 Employees

ZS Logo ZS

Decision Analytics Associate

Artificial Intelligence • Healthtech • Professional Services • Analytics • Consulting
Hybrid
5 Locations
15000 Employees
Hybrid
Mumbai, Maharashtra, IND
289097 Employees

Pfizer Logo Pfizer

Site Reliability Engineer

Artificial Intelligence • Healthtech • Machine Learning • Natural Language Processing • Biotech • Pharmaceutical
In-Office
2 Locations
121990 Employees

Similar Companies Hiring

Legora Thumbnail
Artificial Intelligence • Legal Tech • Software
New York, New York
700 Employees
Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account