Senior / Staff AI Research Engineer, Real-Time Inference
Read straight from RoboForce's careers page, verified 1d ago.Seekless can tailor your résumé to it.
Seekless can write your résumé for this posting first.
Before you apply
Salary$185k – $325k
LocationMilpitas, CA, Milpitas, CA
Work modeon-site
Typefull-time
SeniorityLead
Experience4+ yrs
DepartmentAI
EquityIncluded
Company size51–200 people
PostedNov 5, 2026 · 2d ago
Verified live1d ago
At a glanceSummarised by Seekless from the posting.
Master's degree in Computer Science, Electrical Engineering, or related field with 4+ years of experience, or a PhD degree
Deep expertise in CUDA programming, GPU architecture, and low-level kernel optimization, including custom kernel authoring with tools such as Triton
Hands-on experience with model quantization, pruning, distillation, and deployment using frameworks such as TensorRT, ONNX Runtime, TVM, or Triton
Proficiency in C++ and Python; strong systems programming and performance profiling skills
Experience deploying ML models on edge or embedded hardware (e.g., NVIDIA Jetson, Orin, or equivalent ARM/GPU SoCs)
Familiarity with embodied AI models — VLA, multimodal transformers, or diffusion-based policies — and their inference characteristics
Familiarity with compiler-based optimization pipelines such as XLA, torch.compile, or MLIR for graph-level model acceleration
Understanding of robotics system constraints such as control-loop timing, sensor fusion latency, and memory bandwidth limits on edge SoCs
Publication or production work in efficient deep learning or on-device ML systems
Requires 5 days/week in-office collaboration with the teams
Skills
CUDA
TensorRT
ONNX Runtime
TVM
Triton
C++
Python
NSight
NVIDIA Jetson
Benefits
Stock options
Health insurance
Dental insurance
Vision insurance
401(k)
Visa sponsorship
Green card support
Lunches
Dinners
Why RoboForce
RoboForce is an AI robotics company developing Physical AI–powered Robo-Labor for dull, dirty, and dangerous work. The company’s robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability.
About the role
We are looking for a Senior / Staff AI Research Engineer, Real-Time Inference to make embodied AI practical on the edge. In this role, you will drive the full stack of model optimization — from CUDA kernel engineering to quantization and compression — to deploy high-performance AI models on edge compute platforms powering RoboForce robots in the field.
Responsibilities
•
Develop and optimize inference pipelines for embodied AI models (VLA, perception, world models) targeting real-time execution on edge hardware such as NVIDIA Jetson platforms.
•
Implement CUDA-level optimizations including custom kernels, memory layout tuning, and hardware-aware graph compilation to minimize model latency.
•
Apply and advance model compression techniques — quantization (INT8/FP16/INT4), pruning, distillation, and structured sparsity — to achieve production-grade throughput on constrained devices.
•
Profile and debug end-to-end inference stacks using tools such as NSight, TensorRT, and Triton to identify and eliminate performance bottlenecks.
•
Collaborate with ML research and robotics teams to co-design model architectures that meet real-time control-loop latency requirements.
•
Establish benchmarking frameworks to evaluate model performance across latency, throughput, power consumption, and accuracy tradeoffs on target hardware.
Requirements
•
Master’s degree in Computer Science, Electrical Engineering, or related field with 4+ years of experience, or a PhD degree.
•
Deep expertise in CUDA programming, GPU architecture, and low-level kernel optimization, including custom kernel authoring with tools such as Triton.
•
Hands-on experience with model quantization, pruning, distillation, and deployment using frameworks such as TensorRT, ONNX Runtime, TVM, or Triton.
•
Proficiency in C++ and Python; strong systems programming and performance profiling skills.
•
Experience deploying ML models on edge or embedded hardware (e.g., NVIDIA Jetson, Orin, or equivalent ARM/GPU SoCs).
•
Requires 5 days/week in-office collaboration with the teams.
Bonus Qualifications
•
Familiarity with embodied AI models — VLA, multimodal transformers, or diffusion-based policies — and their inference characteristics.
•
Familiarity with compiler-based optimization pipelines such as XLA, torch.compile, or MLIR for graph-level model acceleration.
•
Understanding of robotics system constraints such as control-loop timing, sensor fusion latency, and memory bandwidth limits on edge SoCs.
•
Publication or production work in efficient deep learning or on-device ML systems.
Benefits
•
Competitive stock options/equity programs.
•
Health, dental, and vision insurance, 401(k) plan.
•
Visa sponsorship and green card support for qualified candidates.
•
Lunches and dinners, a fully stocked kitchen, and regular team-building events.
Compensation: Salary $185,000–$325,000 USD + Bonus + Equity
Legal
The base salary range above represents the expected compensation for this full-time U.S. position. Final compensation will be determined based on role scope, level, location, job-related skills, experience, and relevant education or training, and may fall outside the listed range in exceptional cases.