Gradient Robotics Logo

Gradient Robotics

Founding ML Engineer

Reposted 21 Days Ago
In-Office
Menlo Park, CA, USA
Senior level
In-Office
Menlo Park, CA, USA
Senior level
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.
The summary above was generated by AI

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.

Similar Jobs

20 Days Ago
In-Office or Remote
San Francisco, CA, USA
250K-395K Annually
Expert/Leader
250K-395K Annually
Expert/Leader
Big Data • Blockchain • Internet of Things • Software
Founding ML Performance Engineer will write custom CUDA kernels, optimize end-to-end model inference for sub-50ms latency, drive 10x stack performance (memory bandwidth, kernel fusion, scheduling), implement zero-copy multi-GPU/multi-node memory, own GPU utilization and memory management, profile and benchmark pipelines, and build performance tooling and standards.
Top Skills: CublasCudaCuda KernelsCuteFlash-AttentionInfinibandNcclNvlinkPyTorchPytorch DistributedRayRoceTensorrtTensorrt-LlmTilelangTritonVllm
20 Days Ago
In-Office or Remote
San Francisco, CA, USA
200K-350K Annually
Senior level
200K-350K Annually
Senior level
Big Data • Blockchain • Internet of Things • Software
Founding ML infrastructure engineer to design and scale a GPU compute platform (1,000+ GPU clusters), build multi-tenant scheduling and marketplace infrastructure, own production reliability, architect feature stores/model registries and experiment tracking, and develop resource orchestration and scheduling to optimize throughput, cost, and latency.
Top Skills: AWSAzureCustom SchedulersCuteGCPInfinibandKubernetesNcclRoceSglangSlurmTensorrt-LlmTilelangTritonVllm
11 Days Ago
In-Office
San Francisco, CA, USA
Senior level
Senior level
Artificial Intelligence • Healthtech • Software • Automation
Lead design and implementation of post-training alignment pipelines (SFT, RLHF, RLAIF) using proprietary medical data. Improve model calibration, safety, and clinical correctness via novel techniques, distributed training on GPU clusters, and rigorous evaluation frameworks. Collaborate with co-founders to define research roadmap and ship production-ready ML systems.
Top Skills: Distributed Training FrameworksGpu ClustersHuggingfaceKnowledge DistillationPythonPyTorchReward ModelingRlaifRlhfSftTransformersVercelVertex

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account