Senior AI Engineer — GPU-Driven Agent Systems (Equity)

Socket.dev; Santa Clara (UT); On-site; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Generous benefits

Job summary

NVIDIA is seeking an outstanding High Performance AI Engineer to build groundbreaking agentic systems for the CUDA ecosystem. You will design agent abstractions, GPU-centric runtimes, and compiler- or runtime-driven solutions to accelerate agent planning, tool-use, code generation, and other high-impact AI workloads.

You will collaborate closely with internal NVIDIA software and hardware teams to push the latest developments into NVIDIA products, crafting reproducible evaluation frameworks and

Qualifications

• Bachelor’s degree in CS/EE or related field; MS/PhD preferred.
• 3+ years in AI systems development, foundational models or orchestration.
• Strong C/C++ and Python with solid software engineering fundamentals.

Responsibilities

• Design, build and optimize agentic AI systems for the CUDA ecosystem.
• Co-design agentic solutions with software, hardware and algorithm teams.
• Develop reproducible evaluation frameworks for performance and quality.
• Collaborate across the AI stack from hardware to frameworks and inference.

Skills

C/C++

Python

GPU programming

Deep learning frameworks

Performance optimization

Education

Bachelor's degree in Computer Science or Electrical Engineering

MS or PhD preferred

Tools

CUDA

PyTorch

TensorFlow

Job description

NVIDIA is seeking an outstanding High Performance AI Engineer to build groundbreaking agentic systems for the CUDA ecosystem. You will design agent abstractions, GPU-centric runtimes, and compiler- or runtime-driven solutions to accelerate agent planning, tool-use, code generation, and other high-impact AI workloads.

You will collaborate closely with internal NVIDIA software and hardware teams to push the latest developments into NVIDIA products, crafting reproducible evaluation frameworks and