---
title: 'Senior Inference Optimization ML Engineer at Rhoda AI'
canonical: 'https://feeny.ai/job/senior-inference-optimization-ml-engineer-rhoda-ai-mountain-view-s9kmbmmt2gmk'
type: 'job'
last_seen: '2026-11-04'
---

# Senior Inference Optimization ML Engineer at Rhoda AI

- **Company:** Rhoda AI
- **Location:** Mountain View, CA
- **Employment:** full-time
- **Posted:** 2026-11-01
- **Last confirmed live:** 2026-11-04
- **Apply:** https://jobs.ashbyhq.com/rhoda-ai/d8be2d9b-bd2c-413d-92e8-75cf8112630f

## Job description

At Rhoda AI, we’re building the next generation of generalist intelligent robots. We own the full robotics stack from high-performance hardware and robot systems to the infrastructure and state-of-the-art foundation world models that control our robots. Our robots are designed to be generalists capable of operating in complex, real-world environments and handling long-tail edge cases, made possible by our cutting edge research and end-to-end system design. We've raised over $450M and are investing aggressively in model research, infrastructure, hardware development, and manufacturing scale-up to make generalist robotics a reality. Running large multimodal foundation models for a growing fleet of robots is expensive. We can't rely on one hardware vendor or the most expensive accelerators. We're looking for an Inference Optimization engineer to make Rhoda's models run fast, correctly, and cheaply on a wide range of hardware: AMD Instinct GPUs, Google TPUs, wafer-scale and dataflow accelerators, consumer GPUs like the RTX 5090, and the Jetson/Thor-class compute on our robots. You'll decide which chip runs which workload, and then make sure it runs well there. You'll work closely with the research, robotics, and infrastructure teams.

## What You'll Do

- Own cost-efficient inference end-to-end. Make Rhoda's foundation models (vision, video, language, proprioception, action) faster and cheaper to run, measured in latency, throughput, and cost per inference.
- Port and optimize models for non-NVIDIA and lower-cost hardware. This includes AMD (ROCm, HIP, MIOpen/hipBLASLt, Composable Kernel), TPU (XLA, JAX, Pallas, PyTorch/XLA), wafer-scale and dataflow accelerators (their SDKs and compilers), and consumer Blackwell GPUs such as the RTX 5090 (FP8/FP4, small memory, multi-GPU setups over PCIe).
- Own on-robot inference. Deploy and optimize models on Jetson/Thor-class embedded compute under tight latency, power, thermal, and memory limits, and decide how work is split between the robot and the cloud to meet real-time control requirements.
- Evaluate hardware rigorously. Build cost/performance models (perf/$, perf/W, memory capacity and bandwidth, interconnect) for each chip and model family. Write clear recommendations on which hardware to buy or rent for serving, offline evaluation, and on-robot use.
- Build hardware-portable optimizations. Write and tune kernels that work across vendors where possible (Triton, Pallas, MLIR-based compilers) and vendor-specific ones where the speedup justifies it. Handle attention, KV cache, memory layout, and fused operators on every backend.
- Apply model compression for constrained hardware. Use quantization (INT8, FP8, FP4/MX formats, AWQ/GPTQ), pruning, distillation, and model or tensor parallelism so models fit and perform well on chips with less memory or limited data-type support.
- Keep results correct and consistent across backends. Build tests that check numerical parity, policy quality, and robot-relevant behaviour, so a model gives the same results whichever chip it runs on.
- Build benchmarking and regression infrastructure across hardware. Track latency baselines, throughput curves, and cost per inference for every model version on every target, with automated detection of regressions.
- Work with research engineers to design models that stay efficient and portable from the start, for example by avoiding operators that only exist on one vendor's hardware.

## What We're Looking For

- 2+ years of experience in inference optimization, ML systems, ML compilers, or a closely related field.
- Deep hands-on experience with PyTorch.
- Strong understanding of compute, memory bandwidth, memory capacity, and interconnect bottlenecks, and how they change from one hardware architecture to another.
- Experience with model optimization techniques: quantization (INT8/FP8/FP4, AWQ), distillation, pruning, and compilation (torch.compile, XLA, TensorRT, IREE/MLIR, TVM, or vendor compilers).
- Familiarity with inference serving frameworks and their support for different hardware (e.g. vLLM on ROCm/TPU, SGLang, JetStream, Triton Inference Server).
- Excellent debugging and measurement skills. You can take "this is slow on chip X" or "this costs too much" and turn it into clear bottlenecks, experiments, and validated improvements.
- A high-ownership, pragmatic mindset. You care about the total cost of a solution, not just peak benchmark numbers.

## Nice to Have (But Not Required)

- Experience optimizing inference for diffusion models.
- Production or deep project experience with at least one accelerator stack other than NVIDIA datacenter GPUs: AMD ROCm, Google TPU, wafer-scale and dataflow accelerators, Groq, Tenstorrent, AWS Trainium/Inferentia, Intel Gaudi, or similar, or serious work deploying on consumer GPUs.
- Kernel authoring on more than one backend: HIP or Composable Kernel on AMD, Pallas on TPU, Triton on both NVIDIA and AMD, CUTLASS/CuTe.
- Contributions to open-source ML compilers or hardware backends (PyTorch, OpenXLA, Triton, vLLM, IREE, TVM).
- Experience running inference on wafer-scale systems, TPU pods, or fleets of consumer GPUs.
- Hands-on experience with Jetson/Thor-class platforms (TensorRT on Jetson, JetPack, DLA, power and thermal profiling).
- Experience with multimodal or video model inference (variable-length sequences, packing/bucketing).
- Speculative decoding, continuous batching, disaggregated prefill/decode, or other LLM serving optimizations.
- Experience with streaming or low-latency systems for real-time robot control.
- Experience working with hardware vendors on driver or compiler bugs and early-access silicon.

## Why This Role

- Direct impact on cost and scale. Every gain in perf/$ you deliver lowers the cost of serving models and evaluating new ones, and of putting capable models on many more robots.
- Strategic influence over hardware. Your benchmarks and recommendations will directly shape which accelerators Rhoda buys and builds on, in the cloud and on the robot.
- High ownership on a small, elite team. You'll own the optimization layer that decides how fast and how cheaply our foundation models run in the real world, on every target we use.

## About Rhoda AI

## Company Overview
- **One-liner**: Rhoda AI builds robot foundation models that learn from internet-scale video to enable manipulation-capable robots to generalize in real-world industrial environments.
- **Entity Type**: Private (Seed + Series A; $680M total funding)
- **Headquarters**: Palo Alto, CA, United States
- **Founded**: 2024
- **Founders**: Jagdeep Singh (CEO), Eric Chan (Chief Scientist), Changan Chen (Chief Research Officer), Andrew Wooten, Gordon Wetzstein (Scientific Advisor) [rhoda.ai/team](https://www.rhoda.ai/team)

## Core Business
- **Primary industry**: Robotics, Artificial Intelligence, Hardware
- **Target customers**: B2B – industrial enterprises in automotive, manufacturing, logistics, and e-commerce
- **Mission/purpose**: “Building at the frontier of embodied intelligence” and “Designing intelligence for the real world” – deploying general-purpose robot foundation models that adapt to the variability of commercial and industrial environments. [rhoda.ai](https://www.rhoda.ai/) | [linkedin.com](https://www.linkedin.com/company/rhoda-ai)

## Products & Services
- **Rhoda Robot Platform (Hardware + Software)**: A turnkey robotic system with custom actuators (25kg rated, 40kg peak payload), safety-rated vision, wheel-base, and brakes in every actuator. The platform runs the **Direct Video Action** model, which pre-trains on over a million real-world videos and post-trains on task-specific robot data. Long-context windows enable single-shot learning from human demonstrations without retraining. [rhoda.ai](https://www.rhoda.ai/)
- **FutureVision Intelligence Layer**: A proprietary architecture that handles real-world industrial tasks autonomously by generalizing beyond conventional VLA pipelines. [rhoda.ai](https://www.rhoda.ai/)
- **Deployment Services**: End-to-end solutions for tasks such as returns processing, bearing decanting, container breakdown, and material transport in logistics and e-commerce facilities. [rhoda.ai](https://www.rhoda.ai/)

## Market Standing
- **Valuation**: Not publicly disclosed
- **Key Metric**: Total funding **$680M** – Seed round $67.4M (April 2024), Series A $162.6M (April 2025), and Series A $450M (March 2026) with 4 investors. [linkedin.com](https://www.linkedin.com/company/rhoda-ai)
- **Notable Investors/Partners**: Not individually named in available sources; partners include customers in automotive, manufacturing, logistics, and e-commerce. [rhoda.ai](https://www.rhoda.ai/)
- **Growth Signals**: Headcount growth of **+20% monthly** (LinkedIn reports 55 employees; Built In reports 73 employees – conflicting data). Talent sourced from companies like QuantumScape, Flexiv Robotics, Scale AI, Google, Meta, NVIDIA, and Stanford University. [linkedin.com](https://www.linkedin.com/company/rhoda-ai) | [builtin.com](https://builtin.com/company/rhoda-ai)

## Competitive Advantages
- **Direct Video Action architecture** with web-scale video pre-training gives a strong prior on motion, physics, and dynamics, enabling generalization beyond lab settings.
- **Custom high-performance actuators** (25kg rated, 40kg peak) with safety-rated brakes and vision – designed for 3+ years of continuous operation at rated payload.
- **Long-context in-context learning** allows the robot to perform tasks single-shot from human demonstrations without retraining.
- **Turnkey industrial deployments** that handle ambiguity and variability (e.g., debris, transparent bags, heavy boxes) – tasks customers initially believed could not be automated. [rhoda.ai](https://www.rhoda.ai/)

## Strategic Focus
- Scaling robot intelligence through a pre-training/post-training paradigm, moving from controlled labs to reliable production environments.
- Expanding commercial deployments across automotive, manufacturing, logistics, and e-commerce verticals.
- Pursuing a path to **physical AGI** by combining foundation models with custom hardware for real-world adaptability. [rhoda.ai](https://www.rhoda.ai/)

## Why Work Here
- **Culture**: “Building at the frontier of embodied intelligence” – emphasis on hard problems, first-principles thinking, and category-defining systems. [linkedin.com](https://www.linkedin.com/company/rhoda-ai)
- **Work policy**: Hybrid workspace (employees combine remote and on-site work), with offices in Mountain View and Palo Alto, CA. [builtin.com](https://builtin.com/company/rhoda-ai)
- **Engineering culture**: Strong research focus (Chief Research Officer, Chief Scientist, many research engineers/scientists). Recent job postings include Research Scientist, Robot Perception Engineer, Mechanical Engineer, Fullstack Engineer, and Product Manager. [builtin.com](https://builtin.com/company/rhoda-ai)
- **Notable perks**: Not explicitly listed, but the company is well-funded ($680M) and at an early stage (founded 2024), offering significant growth opportunity and impact.

## Sources
1. [rhoda.ai](https://www.rhoda.ai/) – Company website (product, technology, commercial applications)
2. [rhoda.ai/careers](https://www.rhoda.ai/careers) – Careers page
3. [rhoda.ai/team](https://www.rhoda.ai/team) – Leadership team and co-founders
4. [builtin.com](https://builtin.com/company/rhoda-ai) – Built In profile (employee count, offices, job listings, hybrid policy)
5. [linkedin.com](https://www.linkedin.com/company/rhoda-ai) – LinkedIn company page (funding, headcount, executive list, talent sources)

## Other roles at Rhoda AI

- [Senior System Safety Engineer - Perception](https://feeny.ai/job/senior-system-safety-engineer-perception-rhoda-ai-mountain-view-4wzr6tw3r49a) — Mountain View, CA
- [Senior/Staff TPM - Hardware](https://feeny.ai/job/senior-staff-tpm-hardware-rhoda-ai-mountain-view-1gs3k8kf579p) — Mountain View, CA
- [Senior TPM - Software & Controls](https://feeny.ai/job/senior-tpm-software-controls-rhoda-ai-mountain-view-y3vxhv09ttb9) — Mountain View, CA
- [Senior / Staff Technical Program Manager, AI Research](https://feeny.ai/job/senior-staff-technical-program-manager-ai-research-rhoda-ai-mountain-view-vwmnc9zskvbs) — Mountain View, CA
- [Marketing Communications Senior Associate](https://feeny.ai/job/marketing-communications-senior-associate-rhoda-ai-mountain-view-vmkhppnjbaa4) — Mountain View, CA
- [Software Engineer, Rhoda Platform](https://feeny.ai/job/software-engineer-rhoda-platform-rhoda-ai-mountain-view-118z5qm9f5q0) — Mountain View, CA
- [Software Engineer Intern, Rhoda Platform](https://feeny.ai/job/software-engineer-intern-rhoda-platform-rhoda-ai-mountain-view-3s3s0rkqmbkw) — Mountain View, CA
- [Software Engineer Intern, Inference](https://feeny.ai/job/software-engineer-intern-inference-rhoda-ai-mountain-view-0yz1xmmcxnsq) — Mountain View, CA
- [Technical Program Manager](https://feeny.ai/job/technical-program-manager-rhoda-ai-mountain-view-1vmebb3sybwj) — Mountain View, CA
- [Solution Engineer](https://feeny.ai/job/solution-engineer-rhoda-ai-mountain-view-1fkt9bqkr8g7) — Mountain View, CA