> Markdown version of [/jobs/ext/1893493-ai-infrastructure-engineer](https://www.wearedevelopers.com/jobs/ext/1893493-ai-infrastructure-engineer). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# AI Infrastructure Engineer

- **Company:** Kodiak Robotics
- **Location:** Mountain View, CA, United States
- **Experience:** Expert
- **Salary:** $190,000.0 - $260,000.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, C++ (Programming Language), Cluster Analysis, Profiling, Nvidia CUDA, Computer Programming, Data Files, Shard (Database Architecture), Distributed Computing Environment, InfiniBand, Python (Programming Language), Systems Development Life Cycle, Data Streaming, Scripting, Graphics Processing Unit (GPU), Pytorch, Information Technology, Data Management, Machine Learning Operations, Lidar, Data Pipelines
- **Published:** July 31, 2026
- **Apply:** https://www.careerbuilder.com/job-details/senior-ai-infrastructure-engineer-model-training-mountain-view-ca--5d76d253-c270-425a-976f-a4f774b878cd

## About the Role

* BS, MS, or PhD in Computer Science or a related field, and at least 2-3 years of industry experience in ML systems or infrastructure
 * Hands-on experience with distributed training frameworks and techniques (PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL) and a strong grasp of parallelism trade-offs
 * Experience building high-performance data pipelines for large-scale training, including streaming dataset formats (WebDataset, MosaicML Streaming/MDS, or similar), sharding, and storage/network-aware loading
 * Deep understanding of GPU performance: mixed precision, memory hierarchy, kernel fusion, profiling tools (Nsight, PyTorch Profiler), and interconnects (NVLink, InfiniBand)
 * Strong Python skills and proficiency in PyTorch internals; systems-level experience (C++/CUDA/Triton) a plus
 * Passion for building the infrastructure that lets AI for the physical world train faster, scale further, and improve continuously, Artificial Intelligence (AI), Business Operations, C++ Programming Language, CPU (Central Processing Unit), CUDA (Compute Unified Device Architecture), Cargo/Freight, Computer Science, Construction, Data Clustering, Data Management, Data Modeling, Data Partitioning, Data Sets, Dental Insurance, Disability Insurance, Equal Employment Opportunity (EEO), Flexible Spending Accounts, GPU (Graphics Processing Unit), Government, Health Plan, High Throughput, Incentive Programs, Kernel Programming, Life Insurance, Light Detection and Ranging (LiDAR)\Laser Detection and Ranging (LADAR), Memory Hardware, Patents, Prototyping, Psychiatry and Mental Health, Python Programming/Scripting Language, Regulations, Robotics, Scalable System Development, Team Player, Test Plan/Schedule, Trucking, United States Department of Defense (DoD), Vision Plan



## Related Videos

- [Developing an AI.SDK](https://www.wearedevelopers.com/videos/198-developing-an-ai-sdk)
- [Profiling Symfony & PHP apps with Blackfire](https://www.wearedevelopers.com/videos/265-profiling-symfony-php-apps-with-blackfire)
- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [How to develop an autonomous car end-to-end: Robotic Drive and the mobility revolution](https://www.wearedevelopers.com/videos/22-how-to-develop-an-autonomous-car-end-to-end-robotic-drive-and-the-mobility-revolution)
- [Remote Driving on Plant Grounds with State-of-the-Art Cloud Technologies](https://www.wearedevelopers.com/videos/251-remote-driving-on-plant-grounds-with-state-of-the-art-cloud-technologies)
- [Building the Nervous System of AI - Michael Kagan (NVIDIA)](https://www.wearedevelopers.com/videos/2133-building-the-nervous-system-of-ai-michael-kagan-nvidia)

## Related Articles

- [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)
- [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)
- [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development)
- [What Industries Outside of AI Are Hiring The Most AI Experts?](https://www.wearedevelopers.com/magazine/98-what-industries-outside-of-ai-are-hiring-the-most-ai-experts)
- [Best US AI Conferences for CTOs in 2026: Build vs. Buy, Vendor Evaluation, and Peer Intelligence](https://www.wearedevelopers.com/magazine/736-best-us-ai-conferences-for-ctos-in-2026-build-vs-buy-vendor-evaluation-and-peer-intelligence)
- [Stephan Gillich - Bringing AI Everywhere](https://www.wearedevelopers.com/magazine/489-stephan-gillich-bringing-ai-everywhere)