Senior / Staff AI Research Engineer, Real-Time Inference

Company : RoboForce

Job Type : Full Time

Milpitas, United States

Number of Applicants

000+

Apply Now

Let AI Supercharge Your Job Hunt!

JobCopilot scans 500,000+ company career sites daily to find jobs for you

Never miss an opportunity Save hours by auto-filling applications forms Land more interviews with tailored applications

Activate JobCopilot

Job Description - Senior / Staff AI Research Engineer, Real-Time Inference

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.

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.

Original job Senior / Staff AI Research Engineer, Real-Time Inference posted on GrabJobs ©. To flag any issues with this job please use the Report Job button on GrabJobs.

Apply Now

Auto-Apply to Similar Jobs

Share Job

Get your Resume Reviewed for Free

Automate Job Applications for Similar Jobs

Auto-Apply to AI Research Engineer Jobs with your AI JobCopilot

Auto-Apply with AI

Similar AI Research Engineer Jobs in the US

Get your Resume Reviewed for Free

Email address

Why are you reporting this job?

I think it’s a discriminatory or offensive

I think it’s fraudulent or a scam

I think it’s trying to sell something unrelated to the job / it’s asking for money

I think it contains incorrect or broken information

Other

All Job Ads are subject to GrabJobs’s Terms of Service. We allow users to flag postings that may be in violation of those terms. Job Ads may also be flagged by GrabJobs moderation team. However, no moderation system is perfect, and flagging a posting does not ensure that it will be removed.

Setup your job alert:

Frequency

By activating job alerts, I agree to GrabJobs Terms & Privacy Policy. I can unsubscribe to job alerts anytime. Skip

Senior / Staff AI Research Engineer, Real-Time Inference

Job Description - Senior / Staff AI Research Engineer, Real-Time Inference

Similar AI Research Engineer Jobs in the US

Mobile Apps