High Performance AI Engineer
NVIDIA AI · Santa Clara, CA · Today
EngineeringFull-time
About the role
We are looking for outstanding High Performance AI Engineers to build groundbreaking multi-agent systems for the CUDA ecosystem. Our team builds innovative agentic runtimes and compiler-integrated orchestration that work together with NVIDIA's software stack to provide comprehensive acceleration for modern agent workloads powered by foundational models.
Responsibilities
- Design, build and optimize agentic AI systems for the CUDA ecosystem.
- Co-design agentic system solutions with software, hardware and algorithm teams; influence and adopt new capabilities as they become available.
- Develop reproducible, high-fidelity evaluation frameworks covering performance, quality and developer productivity.
- Collaborate across the AI stack—from hardware through compilers/toolchains, kernels/libraries, frameworks, distributed training, and inference/serving—and with model/agent teams.
Requirements
- Bachelor’s degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD preferred.
- 3 years+ industry or academia experience with AI systems development; exposure to building foundational models, agents or orchestration frameworks; hands-on experience with deep learning frameworks and modern inference stacks.
- Strong C/C++ and Python programming skills; solid software engineering fundamentals.
- Experience with GPU programming and performance optimization (CUDA or equivalent).
Qualifications
- Strong experience in building/evaluating deep learning models, coding agents and developer tooling.
- Demonstrated ability to optimize and deploy high-performance models, including on resource-constrained platforms.
- Demonstrated ability in GPU performance optimizations, evidenced by benchmark wins or published results.
- Publications or open-source leadership in deep learning, multi-agent systems, reinforcement learning, or AI systems; contributions to widely used repos or standards.
Skills
- Strong C/C++ and Python programming skills.
- Experience with GPU programming and performance optimization (CUDA or equivalent).
- Experience with deep learning frameworks and modern inference stacks.
- Experience with building agentic AI systems.
- Experience with developing reproducible, high-fidelity evaluation frameworks.
- Experience with collaborating across the AI stack.
Benefits
- Competitive salaries.
- A generous benefits package.
Pay
- Base salary range: $124,000 - $195,500 for Level 2, and $152,000 - $241,500 for Level 3.
Schedule
- Full-time position.