NVIDIA
Senior Software Engineer, AI Frameworks
$152K–$288K
US, CA, Santa Clara +1 · Senior · Full-time · Posted Oct 29, 2026
- Location
- US, CA, Santa Clara
US, CA, Remote - Experience
- 5+ years of proven experience in a related field.
- Education
- BS/MS/PhD in Computer Science, Electrical Engineering, or related field (or equivalent experience).
About the job
Integrate NVIDIA Grove with open-source AI frameworks and build production-grade software, tooling, and workflows for distributed training and inference.
What you'll do
- Design, implement, and maintain Grove integrations, adapters, plugins, and runtime components across AI frameworks.
- Optimize performance, scalability, reliability, and observability for distributed deployments; diagnose issues across the GPU and cloud-native stack.
- Collaborate with framework owners and open-source communities on upstream contributions, reference workflows, APIs, and documentation.
What they ask for
- Hands-on integration experience with at least one major AI framework or runtime.
- Understanding of AI workloads, training and inference tradeoffs, and performance considerations.
- Experience with distributed systems concepts and practical Kubernetes operations.
- Software engineering experience in Go, C++ and/or Python, with a record of shipping reliable systems.
Nice to have
- Open-source contributions to Dynamo, PyTorch, Ray, llm-d, Kubernetes ecosystem, or related ML infrastructure projects.
- Experience in large-scale model serving, GPU optimization, developer tooling, or compatibility testing.
Pay
152,000 USD - 241,500 USD base salary for Level 3; 184,000 USD - 287,500 USD base salary for Level 4
Tools
Dynamo, llm-d, Ray, PyTorch, Kubernetes, Helm/Kustomize, Docker, Go, C++ and/or Python
AI focus
Enables adoption and operation of Grove across AI frameworks and supports distributed training and inference.
Model-written summary · the original posting is authoritative