Founding ML Engineer

Reposted One Month Ago
Menlo Park, CA, USA
In-Office
Senior level
Artificial Intelligence • Robotics • Industrial • Automation
The Role
Train, fine-tune, and deploy vision-language-action and robot manipulation models across the full pipeline. Handle data curation, pre-training and fine-tuning, sim-to-real distillation, debugging, evaluation, and iterative autonomous improvement on physical robots.
Summary Generated by Built In

About Gradient Robotics

Gradient Robotics is building intelligent robots for data centers and factories. We believe the fastest practical path to capable general-purpose robotics is through industry deployment. We're here to change the world with robotics.

We own and build our full robotics stack — designing the hardware, building the software systems, training ML models, and manufacturing through a global supply chain — for world-class iteration speed, reliability, and scale.

We've built 3 generations of robots with a team of 7, backed by top-tier investors and working toward pilots with the world's largest data center builders. Our team created and built the best-selling open-source humanoid robots in the US at K-Scale Labs and worked on generations of foundation software for Tesla Optimus.

The Role

As a founding ML engineer at Gradient, you'll train and deploy the models that make our robots useful in the real world. You'll work across the full learning pipeline — from data collection and curation through VLA pre-training and fine-tuning to RL-based self-improvement on real hardware. Your models won't live in a paper. They'll fold boxes, move pallets, and operate in environments they've never seen before.

What You'll Do

  • Train and fine-tune vision-language-action models for manipulation on Gradient's robots

  • Work on distilling policies trained in sim environments to real environments

  • Constantly assess data quality and setup pre-training + post-training split to fine-tune the backbone and action heads

  • Debug training failures, identify failure modes, and close the loop between model performance and data collection strategy

  • Own the full cycle: data → training → eval → deployment → autonomous improvement

Background

  • Deep Learning: PyTorch or JAX, distributed training, mixed precision

  • Robot Learning: Imitation learning, diffusion policies, flow matching, or VLA architectures (π₀, OpenVLA, ACT, RT-X, Octo, etc.)

  • Reinforcement Learning: Sim-to-real transfer, distillation techniques

  • Simulation: MuJoCo, Isaac Sim, or Genesis

  • Python: Comfortable writing research-grade code that ships to production

Nice to Have

  • Experience training or fine-tuning large VLMs (PaLI, Gemma, LLaVA)

  • Multi-modal data pipelines at scale (video, proprioception, language, tactile)

  • Publications at CoRL, RSS, ICRA, NeurIPS, ICML, or ICLR

  • Experience deploying learned policies on physical robots

At Gradient, your models ship to real robots daily. You'll watch them get smarter, more general, and more reliable as you close the loop between learning and deployment. If that excites you, we should talk.

Skills Required

  • Strong deep learning experience (PyTorch or JAX), distributed training, mixed precision
  • Robot learning knowledge: imitation learning, diffusion policies, flow matching, or VLA architectures
  • Reinforcement learning experience including sim-to-real transfer and distillation techniques
  • Simulation experience with MuJoCo, Isaac Sim, or Genesis
  • Proficient Python skills; able to write research-grade code that ships to production
  • Experience training or fine-tuning large VLMs (PaLI, Gemma, LLaVA)
  • Experience building multi-modal data pipelines at scale (video, proprioception, language, tactile)
  • Publications at CoRL, RSS, ICRA, NeurIPS, ICML, or ICLR
  • Experience deploying learned policies on physical robots
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
Year Founded: 2025

What We Do

The datacenter buildout is the largest industrial project in human history. Gradient builds the autonomous robots that make it possible. Partnered with the world's largest AI infrastructure companies and backed by the industry's best investors, we move fast and build full-stack systems that matter.

Similar Jobs

FilmoreAI Logo FilmoreAI

Founding Engineer, AI & ML Systems

Artificial Intelligence • Machine Learning • Software • Analytics
In-Office or Remote
2 Locations
3 Employees

Autostep Logo Autostep

Founding Engineer, ML + Full-Stack

Artificial Intelligence • Machine Learning • Software • Automation
In-Office
San Francisco, CA, USA
2 Employees

HowURun Logo HowURun

Founding Engineer - ML Performance

Big Data • Blockchain • Internet of Things • Software
In-Office or Remote
San Francisco, CA, USA
250K-395K Annually

HowURun Logo HowURun

Founding Engineer - ML Infrastructure

Big Data • Blockchain • Internet of Things • Software
In-Office or Remote
San Francisco, CA, USA
200K-350K Annually

Similar Companies Hiring

Revel.io Thumbnail
Aerospace • Hardware • Robotics • Software
US
50 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account