7 days ago
Santa Clara, CA, USASenior / Staff+
H1B Sponsor
Base Salary
$152k - $288k/yr
Responsibilities
- Architect and optimize end-to-end training workflows for robotics foundation models with researchers and ML engineers.
- Build proof-of-concepts, reference architectures, and agentic workflows for experimentation, benchmarking, and model improvement.
- Scale pre-training, fine-tuning, and reinforcement-learning workloads across multi-GPU and multi-node systems.
- Identify and eliminate storage, networking, preprocessing, and data-loading bottlenecks for multimodal datasets.
- Collaborate with NVIDIA product and engineering teams to influence future Physical AI platforms.
- Profile and optimize training workloads and support model evaluation, data analysis, and research acceleration.
Requirements
- MS, PhD, or equivalent experience in Computer Science, Artificial Intelligence, Electrical or Computer Engineering, Robotics, or a related field.
- 5+ years of industry or research experience in deep learning, distributed computing, or large-scale model training.
- Hands-on experience training or optimizing multimodal or foundation models, ideally for robotics.
- Experience across pre-training, supervised fine-tuning, reinforcement learning or other post-training methods, evaluation, and model optimization.
- Strong expertise in distributed training techniques including data, model, and pipeline parallelism, sharding, and checkpointing.
- Expertise with PyTorch, NVIDIA NeMo, JAX, or Hugging Face Transformers.
- Experience building or using high-throughput data pipelines for large-scale training.
- Strong communication and cross-functional collaboration skills.
- Familiarity with NVIDIA AI and robotics platforms, robotics reinforcement learning and synthetic data generation, workload profiling, and agentic experimentation workflows is preferred.
Benefits
- Eligible for equity and benefits.
- Base salary varies by location, experience, and comparable employee pay.
- Applications will be accepted at least until August 31, 2026.
Tech Stack
Hugging Face TransformersPyTorch
Categories
Solutions Engineering
About Nvidia
Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.