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.

- Solid 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.

📌 Machine Learning Engineer, Robotic Foundation Models (Bengaluru)
🏢 MetAntz
📍 Bengaluru