Senior AI Infrastructure Engineer, AI Tooling - DGX Cloud

Details

We are looking for a Senior AI Infrastructure Engineer (AI Tooling) to design and build the backend systems and infrastructure powering our internal AI tools and intelligent agents. This is a hands-on software engineering role with a strong infrastructure focus, ideal for someone who thrives at the intersection of backend development, systems engineering, and applied AI.

You will contribute to building robust, scalable systems that support a wide range of applied AI initiatives. These may include integrating LLMs via APIs, supporting internal ML models, working with AI agent frameworks, building data pipelines, and developing tools to improve AI reliability, observability, and deployment workflows. This is not a research or model training role. It is a practical engineering position focused on enabling and scaling real-world AI applications through solid infrastructure and backend development.

This is a full-time role. Applications for this job will be accepted at least until December 18, 2025.

Responsibilities

• Build AI-powered tools that enhance operational excellence by leveraging diverse operational data, supporting incident, change, and problem management workflows.
• Collaborate closely with Incident Commanders, incident response, and SRE teams to integrate AI-driven automation and analytics into operational workflows.
• Design, develop, and maintain backend systems and infrastructure using Go and Python to support internal AI tools and intelligent agents.
• Build and maintain data pipelines using PySpark and related tools to support AI and analytics workflows.
• Operate and scale infrastructure using Kubernetes, managing containerized AI services and automating pipeline deployments.
• Work with vector databases to enable semantic search and retrieval-augmented generation use cases.
• Integrate large language models, agent systems, and classical ML models into internal services and automation workflows.
• Improve observability, deployment automation, and system reliability for AI-driven services.

Requirements

• 8+ years of software engineering experience, with deep expertise in backend systems and infrastructure.
• Bachelor's degree or equivalent experience.
• Strong proficiency in Python and Go, with a track record of delivering reliable, scalable software solutions.
• Experience designing scalable, maintainable backend systems and writing clear design documentation.
• Deep experience with Kubernetes and cloud-native infrastructure.
• Experience building, deploying, and maintaining ML models in production systems.
• Familiarity with AI agent frameworks or orchestration tools.
• Solid understanding of system observability, monitoring, and performance optimization.
• Strong collaboration and communication skills to work across remote teams.

Ways to stand out from the crowd

• Proven experience delivering AI-driven features end-to-end, from infrastructure design to production deployment.
• Experience with Temporal, Argo, or other workflow engines for multi-step or async job orchestration.
• Familiarity with vector databases and their applications in AI and LLM-powered systems.
• Contributions to open-source projects related to AI infrastructure, backend development, or developer tooling.

Compensation & Benefits

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5. You will also be eligible for equity and benefits (see company benefits page).

Equal Opportunity

NVIDIA is committed to fostering a diverse work environment and is proud to be an equal opportunity employer. We do not discriminate on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.