Software Engineer, Agent Performance Modeling - New College Grad 2026
$124k - $242k • Remote • US, CA, Santa Clara
Posted 2d ago
Remote Work Policy
Fully remote
Categories
AI Agent Engineer
About the job
NVIDIA is seeking to define the next era of computing by leveraging the unlimited potential of AI, with GPUs acting as the core intelligence for advanced systems. This role focuses on understanding how evolving LLM architectures and inference workloads map to hardware, which is critical for designing efficient AI systems. The team utilizes simulation and agentic engineering to analyze emerging architectures and inform hardware and software development decisions, aiming to make a lasting impact on the world.
Responsibilities
- Build and validate performance models for LLM architectures across various systems.
- Develop agents to automate simulation workflows, identify bottlenecks, and speed up design exploration.
- Collaborate with cross-functional teams to enhance modeling accuracy and guide optimizations.
Requirements
- Recently completed MS in Computer Science or related field, or equivalent experience.
- Familiarity with modern LLM architectures and their inference characteristics.
- Experience with simulation, profiling, or performance modeling, including roofline analysis for LLM inference.
- Strong analytical skills and a rigorous approach to experimentation and validation.
- Experience building reliable engineering tools using coding agents.
Benefits
- Equity
- Benefits