• Companies
  • News
By technology
  • Python17.1k
  • AWS11.9k
  • Kubernetes8.5k
  • Docker8.2k
  • C++5.7k
  • PostgreSQL5.4k
  • SQL5.1k
By category
  • Backend Developer11.5k
  • DevOps Engineer3.5k
  • Machine Learning Engineer3.3k
  • Full Stack Developer3.2k
  • Data Engineer2.8k
  • Embedded Engineer2.2k
  • Frontend Developer863
By city
  • San Francisco, CA6.2k
  • New York, NY5.6k
  • Seattle, WA1.3k
  • Washington, DC1.2k
  • Toronto, ON1.2k
  • Boston, MA1k
  • Austin, TX1k
Work mode
  • On-site22k
  • Remote10.2k
  • Hybrid7.3k
Job type
  • Full-time38.2k
  • Contract601
  • Part-time529
  • Freelance215
  • internship4
  • seasonal1
Experience
  • Entry level
  • Mid level
  • Senior
  • Staff+
  • Engineering management

The job board for developers. Find your next role at companies that value great engineering.

© 2023-2026 . All rights reserved.

Logos provided by Logo.dev
Add
8 matches
NewestHighest base
Zoox

Software Engineer, ML Training Optimization

today
Zoox · Foster City, CA
$177k–270kOn-siteFull-timePyTorch · Megatron-LM
Zoox

Senior/Staff Software Engineer, ML Training Optimization

today
Zoox · Foster City, CA
$254k–350kOn-siteFull-timeMegatron-LM · TorchTitan
Percepta

Research Scientist – Reinforcement Learning (RL)

today
Percepta · New York, NY
$200k–400kOn-siteFull-timePython · Kubernetes
Krea

ML Researcher - Posttraining

today
Krea · San Francisco, CA
On-siteFull-timePyTorch · vLLM
Crosby

Member of Technical Staff, Research Engineer

today
Crosby · New York, NY
$200k–300kOn-siteFull-timePython · vLLM
Ema

AI Resident

today
Ema · San Francisco, CA
$48k–48kHybridFull-timePython · PyTorch
Modal

Forward Deployed Engineer - ML

today
Modal · New York, NY
$180k–250kOn-siteFull-timevLLM · SGLang
Baseten

Software Engineer - Training Product

today
Baseten · San Francisco, CA
$165k–330kHybridFull-timeKubernetes · PyTorch
Zoox

Software Engineer, ML Training Optimization

Zoox · 🇺🇸 Foster City, CA · posted today
ApplyFull posting

Extracted from the posting

Base
$177,000 – $270,000
Remote
On-site
Stack
PyTorch, Megatron-LM, TorchTitan, NCCL

TL;DR

Software Engineer on the ML Platform team building training infrastructure for Zoox's foundation, VLM, and VLA models used in autonomous driving. Work covers distributed training optimization (data/tensor/pipeline parallelism, mixed precision, GPU kernels), RL performance, and profiling across compute, compiler, and cluster networking. Stack referenced includes Megatron-LM, TorchTitan, NCCL, Verl/MILES, Python or C++, and profiling tools such as Nsight and PyTorch Profiler.

About Zoox →

The full posting

Zoox is on a mission to reimagine transportation and ground-up build autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone. We are still in the early stages of deploying our robotaxis on public roads, and it is a great time to join Zoox and have a significant impact in executing this mission. The ML Platform team at Zoox plays a crucial role in enabling innovations in large-scale Foundation models, VLMs, and VLAs to make autonomous driving as seamless as possible.

The Opportunity

Are you excited to drive our ML Training Optimization initiatives and make the models that power autonomous driving trains faster, cheaper, and at greater scale? You will work with SOTA accelerators and cutting-edge techniques in distributed training, parallelism strategies, mixed-precision training and RL, especially for VLA and VLM models that are multi-modal — partnering closely with all the Autonomy teams within Zoox (Perception, Prediction, Planner, Simulation, Collision Avoidance) and pushing the boundaries of how ML training is practiced at Zoox.

We build and operate the base layer of ML tools, model development, and training systems that our applied research teams use for in- and off-vehicle ML use cases. You will work alongside a team of strong software engineers and act as a force multiplier for our internal customers. This team has many growth opportunities as we expand our robotaxi deployments and venture into new ML domains. If you want to learn more about our stack behind autonomous driving, please look

here

. If you want to learn more about our ML Infrastructure, here is one of our

past talks

at re:Invent.

In this role, you will:

    Qualifications

      About Zoox

      Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.


      Follow us on LinkedIn


      Accommodations

      If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.


      A Final Note:

      You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.

      Compensation: USD 177,000 – 270,000 per year salary