I

Senior ML Infrastructure Engineer

inventure • United States
Visa Sponsorship
This Job is No Longer Active This position is no longer accepting applications

Architect and scale distributed training infrastructure for large-scale multimodal robotic AI models. Build high-throughput data pipelines and low-latency inference systems to support real-time robot control. Requires 5+ years of experience in ML infrastructure, PyTorch, and distributed training frameworks.

Key Highlights
Own end-to-end training infrastructure for a robotics company with $140M+ funding.
Work on-site in Redwood City, CA, with visa sponsorship available.
Competitive base salary of $220K–$350K plus equity.
Key Responsibilities
Architect and scale distributed training across large GPU clusters using sharding, activation checkpointing, and memory optimization.
Build researcher-friendly tooling and job scheduling on Kubernetes and SLURM with automated retries and failure recovery.
Design high-throughput data pipelines to ingest and transform terabytes of multimodal robot data.
Build low-latency inference pipelines for real-time robot control using quantization, distillation, and model compilation.
Profile deep into the stack to optimize GPU utilization, I/O bottlenecks, and memory fragmentation.
Technical Skills Required
PyTorch Distributed Training Kubernetes
Benefits & Perks
Competitive equity
Visa sponsorship (OPT and H1B transfers)
Nice to Have
Robotics experience
Experience building multimodal systems for video, audio, or other rich media models

Job Description

ML Infrastructure Engineer | Redwood City, CA (Onsite) | $220K–$350K base

Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources

Similar Jobs

Explore other opportunities that match your interests

Principal Architect, AI Factory

Devops
•
11h ago

Log in free to see full details

•••••• •••••• ••••••
Technical Skills Required ••••••

Caterpillar Inc.

United States

Applied Researcher 4 - AI Foundations

Devops
•
15h ago

Log in free to see full details

•••••• •••••• ••••••
Technical Skills Required ••••••

Capital One

United States

Senior AI Cloud Security Engineer

Devops
•
18h ago

Log in free to see full details

•••••• •••••• ••••••
Technical Skills Required ••••••

Deloitte

United States

Visas, Work Permits & Legal Insights

Stay informed about visa types, work permits, and immigration updates for international professionals.

Browse all articles

Subscribe our newsletter

New Things Will Always Update Regularly