Jobs in the United States

25+ jobs

Group jobs by employerEmail me new jobs Save this search

Added 2 mins ago
Cyber Data Analysis Engineer III
CACI
Springfield, VA · Springfield
A Cyber Data Analysis Engineer designs, builds, and maintains the enterprise environment for collecting, storing, and displaying data that supports cybersecurity efforts. In this role,…
Added 2 mins ago
Network Engineer III
CACI
Springfield, VA · Springfield
Team CACI has an exciting and challenging opportunity available for a Network Engineer who performs research, requirements analysis, design, installation, documentation, troubleshooting,…
Added 5 mins ago
Process Engineer
Reynolds Consumer Products
Appleton, WI
Partner with Production, Maintenance, Quality, Research and Development, and Engineering to resolve technical issues, coordinate corrective actions, and improve plant performance.
Added 5 mins ago
Senior Systems Engineer I
Elsevier Inc. Company
CA
About the role; As a Sr Systems Engineer you will support our AWS Cloud infrastructure, automation, and operational tasks. Working in partnership with product and engineering teams, this…
Added 6 mins ago
Sr AI Engineer I - Enterprise Architecture
American Express
Phoenix, AZ
Our focus is on agentic AI development: designing intelligent, adaptive systems that can plan, reason, and act across complex workflows with appropriate levels of autonomy. These systems…
Added 7 mins ago
Senior Test Engineer - CST - Lafayette, CO (Onsite)
SNT Medtronic Navigation, Inc.
Lafayette, CO
Cranial & Spinal Technologies develops integrated technologies and procedural solutions that support the diagnosis and treatment of neurological and spinal conditions. Within Research and…
Added 7 mins ago
G
Cybersecurity Engineer - IT Security - Full Time
Guthrie
Remote, United States
- Full-time
…you are responsible for defending the enterprise from threats and adversaries, both internal and external. The Cybersecurity Engineering team leads security technology research,…
Added 9 mins ago
Reliability Engineer - Lilly Medicine Foundry
Lilly
Lebanon, IN · Indianapolis, IN
The Reliability Engineer increases the reliability of process and utility equipment in support of manufacturing research requirements for capacity, dependability, quality, safety,…
Added 10 mins ago
R&D Electric Motor Drives Engineer
Nexteer Automotive Corporation
Saginaw, MI
We are looking for a highly self-motivated individual who is ready to take on a technology development and innovation role. The candidate should possess good attention to detail, be…
Added 15 mins ago
Senior Principal Systems Engineer
General Dynamics Mission Systems, Inc
Dedham, MA
As a Systems Engineer, working with custom software built in-house, for our Cyber group, you’ll be amember of a cross functional team responsible for developing mission critical…
Added 15 mins ago
BI
Senior Mechanical Engineer III
Bonnell Industries Inc
Dixon, IL
- $80,000 to $90,000 USD / year
Under the direction of the Engineering Manager, the Senior Mechanical Engineer III is responsible for carrying out the duties of the Engineering department, including project management,…
Added 18 mins ago
UG
Senior Software Engineer - 2401660
UnitedHealth Group
Eden Prairie, MN
- $136,080 to $195,174 USD / year
DUTIES: Research, design, and develop computer and network software or specialized utility programs; drive key architecture and infrastructure initiatives with various networking, cyber…
Added 19 mins ago
Client Solutions Engineer
Instem
Conshohocken, PA
Must have ability to travel internationally (maximum of 20% of time)
Added 20 mins ago
Structural Engineer
Vertiv
Pelzer, SC
Must have the ability to research new designs, technologies and construction methods of data center equipment and facilities · Must possess excellent communication skills both written and verbal
Added 21 mins ago
Lead Mechanical Engineer
Concord Engineering Group Inc
Voorhees, NJ
CONCORD ENGINEERING GROUP, INC. a full-service MEP engineering, commissioning, construction management and energy consulting firm with offices in Voorhees, Linwood and NYC, is searching for…
Added 24 mins ago
Embedded Software Development Principal Engineer P4 (Onsite)
RTX
United States
Qualifications You Must Have:
Added 24 mins ago
PSCV Engineering - Validation Engineering - Engine Test & Assembly (Onsite)
RTX
United States
Qualifications You Must Have
Added 24 mins ago
Principal Electrical Engineer - Depot Hardware Lead
RTX
United States
Qualifications You Must Have
Added 24 mins ago
Manufacturing Engineer I (Onsite)
RTX
United States
Qualifications You Must Have
Added 24 mins ago
System Engineer II Onsite
RTX
United States
Qualifications You Must Have
Added 24 mins ago
Software Engineering Intern (ISET-IN-2)
RTX
United States
Qualifications You Must Have
Added 24 mins ago
Engineering Execution Engineer
RTX
United States
Qualification Must Have
Added 24 mins ago
Software Engineer I (CMA-P1-1)
RTX
United States
Qualifications You Must Have
Added 25 mins ago
Staff Motion Planning Engineer
Motional
Remote, United States
On our Motion Planning team, you'll apply your passion in robotics and software development for production-ready autonomous vehicles. In this role you'll perform cutting-edge research,…
Added 32 mins ago
Software Engineering Manager II, TPU Superpod Health Management
Google
Sunnyvale, CA
We're the driving force behind Google's groundbreaking innovations, empowering the development of our cutting-edge AI models, delivering unparalleled computing power to global services, and…

Rhoda AI

Senior Inference Optimization ML Engineer

Mountain View,, CA, United States

Apply on jobs.ashbyhq.com About employer

Applications open in a new tab.

Job details

Experience: 2+ years of experience in inference optimization, ML systems, ML compilers, or a closely related field.
Posting language: English
Posted: Nov 1, 2026

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

Apply on jobs.ashbyhq.com About employer

Report this job

Work requirements

Jobs in the United States