Machine Learning Engineer, Robotic Foundation Models
SproutsAI
Opens the application on app.sproutsai.com
Job Description & Responsibilities
Machine Learning Engineer, Robotic Foundation Models
About MetAntz
MetAntz has one of the most advanced GenAI recruitment platforms. We help companies hire faster with an end-to-end product that covers job posting, sourcing, scoring and matching, and AI-driven assessments. We also run a candidate marketplace and a Recruitment as to Service team that fills roles for our customers. In this role you will sell both the platform and the services around it.
About the role
A machine learning engineer works across the full loop that makes a robot better: data, models, training, simulation, evaluation, and deployment on real hardware. You’ll do research-grade work and ship it onto robots in the same quarter. A result that doesn’t run on the robot isn’t finished.
What you’ll work on
- Training and post-training: Train and post-train robotic foundation models, including vision-language-action models, using imitation learning and reinforcement learning. You’ll own the work from data loading through multi-GPU training runs.
- Training infrastructure: Build and scale distributed multi-node GPU training, data loaders that keep GPUs fed, and reproducible pipelines with experiment tracking.
- Simulation: Build simulation environments for our tasks. Train policies with reinforcement learning that transfer to real robots.
- Rewards and verifiers: Design the reward functions and the verifiers that check whether a task step actually succeeded.
- Evaluation: Build harnesses that tell us whether a model actually got better, in simulation and on real hardware. Design the metrics that decide what ships.
- Data and data infrastructure: Define what data the models need. Build the pipelines that turn raw robot episodes into versioned, training-ready datasets. Trace a stuck policy back to its data.
- Models on robots: Get models running reliably and fast enough on real hardware. Debug failures in the lab across the model, the data, and the robot.
What we’re looking for
- 5+ years of hands-on experience training and shipping ML models in production or in a research lab, with a B.Tech, M.Tech or PhD in Computer Science, Electrical Engineering or a related field.
- A track record of owning an ML system end to end: you’ve taken something from problem definition through data, training, evaluation, and deployment, and you can tell us what you’d do differently.
- Deep hands-on experience with PyTorch, including writing and debugging non-trivial training loops.
- Experience training at scale: multi-GPU or multi-node training (DDP, FSDP, DeepSpeed, or equivalent) and the profiling work to keep it efficient.
- Strong fundamentals in linear algebra, probability, optimization, and machine learning, and in data structures, algorithms, and systems.
- Strong software engineering habits: you write code others can build on, and hold a high bar for code quality.
Preferred
- Experience post-training transformers, diffusion models, or multimodal models, especially with imitation learning or reinforcement learning.
- Experience in robotics, autonomous vehicles, or another domain where models act in the physical world and failures are expensive.
- Experience with robot simulation (MuJoCo, Isaac Sim, PyBullet) and sim-to-real transfer.
- Experience building data pipelines for large multimodal datasets (object storage, Ray or Spark, Parquet, WebDataset).
- Experience deploying models to constrained or real-time hardware (ONNX, TensorRT, quantization, latency work).
- Publications at top AI or robotics venues, or meaningful open-source contributions.
- Experience mentoring engineers or leading a small technical effort.
What we value
- Ownership. You finish things. A result that doesn’t run on the robot isn’t finished.
- Rigor. You catch the subtle bug that the dashboard says is fine.
- Range. You’re as willing to debug a data pipeline or a ROS node as a training run.
- Comfort with real hardware. Our models run on robots, and you’ll spend time in the lab.
- Speed of learning. There’s no playbook. You’ll be learning new parts of the stack every month.
- Ambition. You want to build something that matters at the frontier of AI and robotics.
About SproutsAI
Required Skills
Job Details
Posted by
N/A
Posted on:
22 Oct 2026
About SproutsAI
More open roles
- Data Scientist Machine Learning & Applied Analytics. (Bengaluru)Tata Elxsi · Bengaluru, Karnataka, India
- Vice President - Machine Learning EngineerNatWest Group · Bengaluru, Karnataka, India
- Associate, Machine Learning and AI ApplicationsQuantmatrix AI Solutions · Mumbai, Maharashtra, India
- Machine Learning and Statistical ModellingTekskills Inc. · Kolkata, West Bengal, India
- Senior SQL Machine Learning EngineerInformation Technology AR Robotics India Pvt Ltd · New Delhi, Delhi, India
- Machine Learning Researcher (GPT Transformers SWL)IT Solutions Inc · Hyderabad, Telangana, India