Description
Join the new Agentic Engineering team within the Deep Learning Framework Group at NVIDIA. We build agentic workflows that automate code generation, testing, and tuning across NVIDIA's frameworks, compilers, and developer tooling. This greenfield opportunity offers foundational technical influence within a high-autonomy team.
You will develop a deep understanding with early-adopter engineering teams, identifying friction points where agentic workflows would have the highest impact. As an applied ML expert, you will use technical judgment to distinguish durable architectural opportunities from hype.
The work spans several areas, including:
- Agent-ifying compiler infrastructure for autonomous agents to make high-dimensional optimizations
- Multi-agent orchestration using frameworks like LangChain/LangGraph
- Integrating systems into git-native workflows and CI pipelines
Requirements:
- MS in Computer Science, Engineering, or equivalent experience
- 12+ years of experience
- Strong Python development skills
- Working knowledge of GPUs or other highly data-parallel systems
- Demonstrated projects or work experience using and supporting AI systems
- Track record of shipping complex projects with minimal direction
- Experience building tools or systems shaped by direct partnership with internal customer or user teams
Experience in one or more of the following areas is desirable:
- Multi-agent orchestration frameworks (e.g., LangChain, LangGraph) or LLM-based workflow automation
- Compiler infrastructure, intermediate representations, or program transformation
- Autonomous search or optimization over high-dimensional parameter spaces
- Hardware-aware performance optimization for deep learning workloads
- Code generation systems or domain-specific languages (DSLs)
To stand out:
- Passion for following the evolution of ML hardware and emerging kernel programming techniques
- Experience building evaluation or testing harnesses for ML systems or multi-agent workflows
- Track record of building internal tools or frameworks that force-multiply engineering teams
- Demonstrated ability to thrive in ambiguous, self-directed environments
The base salary range is 224,000 USD - 356,500 USD. You will also be eligible for equity and benefits.