Software Engineer, Inference Systems

River AI; Palo Alto (CA); On-site; USD 200,000 - 420,000; Full time

Benefits offered by this job

Health insurance

Dental insurance

Vision insurance

Unlimited PTO

Relocation assistance

Visa sponsorship

Job summary

River AI in Palo Alto is seeking exceptional inference systems engineers to build the engines that serve large models through the River API. Your goal is to deliver fast, reliable inference while making efficient use of GPU compute and memory.

You will own the serving runtime from request scheduling and continuous batching to KV-cache management, distributed model execution, and checkpoint loading. Working closely with GPU kernel engineers, researchers, and infrastructure engineers, you will

Qualifications

• Bachelor’s degree in CS, CE, or equivalent practical experience.
• Experience building inference engines or performance-sensitive distributed services.
• Strong knowledge of transformer inference, GPU memory, concurrency, and networking.
• Proficiency in Python and C++ or Rust.
• Strong debugging and profiling skills across models, runtimes, and services.
• Collaborative mindset with ownership of outcomes.

Responsibilities

• Optimize inference for dense and mixture-of-experts models.
• Improve batching, caching, and admission control for throughput and latency.
• Accelerate multi-GPU execution while preserving model-version consistency.
• Improve RL sampling throughput with correct model versioning.
• Build reliable streaming, cancellation, and recovery under load.
• Profile bottlenecks and validate improvements with reproducible tests.

Skills

Python

C++

Rust

Transformer inference

GPU memory

Concurrency

Networking

Debugging

Profiling

Distributed services

Education

Bachelor's degree in CS/CE

Tools

SGLang

vLLM

TensorRT-LLM

CUDA

NVIDIA profiling tools

Job description

At River AI, our mission is to create personal AI owned and shaped by each individual. To achieve this, we are rewriting the entire stack from scratch: personal hardware for local inference, bespoke training infrastructure, next-generation UIs, and frontier deep learning research.

Who we are

We are scientists, engineers, and builders from the industry's top tech companies and AI labs. We bring a proven track record of scaling consumer systems for hundreds of millions of users and architecting the pre-training infrastructure behind today's frontier models.

About The Role

We are looking for exceptional inference systems engineers to build the engines that serve large models through the River API. Your goal is to deliver fast, reliable inference while making efficient use of GPU compute and memory.

You will take ownership of the serving runtime, from request scheduling and continuous batching to KV-cache management, distributed model execution, and checkpoint loading. Your work will support both customer-facing inference and the sampling workloads that power reinforcement learning.

Working closely with GPU kernel engineers, researchers, and infrastructure engineers, you will bring new models into production and improve their performance across realistic workloads. You will measure success through latency, throughput, reliability, and cost, with careful attention to numerical correctness and model behavior.

What You’ll Do

• Optimize inference for dense and mixture-of-experts models, including fine-tuned models and adapters.
• Improve batching, caching, and admission control to balance throughput, latency, memory use, and fairness.
• Accelerate multi-GPU execution, communication, and model loading while preserving model-version consistency.
• Improve RL sampling throughput while keeping samples and log probabilities tied to the correct model version.
• Build reliable streaming, cancellation, and recovery under failures and overload.
• Profile bottlenecks and validate improvements through reproducible performance and correctness tests.

Skills & Qualifications

Minimum Qualifications

• Bachelor’s degree in Computer Science, Computer Engineering, or equivalent practical experience.
• Experience building inference engines or performance-sensitive distributed services.
• Strong understanding of transformer inference, GPU memory, concurrency, and networking.
• Proficiency in Python and C++ or Rust.
• Strong debugging and profiling skills across models, runtimes, and services.
• A collaborative mindset and strong ownership of engineering outcomes.

Preferred Qualifications

(We encourage you to apply even if you don't meet all of these)

• Experience extending SGLang, vLLM, TensorRT-LLM, or similar frameworks.
• Work on advanced serving techniques, such as speculative decoding or disaggregated prefill and decode.
• Familiarity with expert parallelism, GPU collectives, and quantized inference.
• Experience with multi-adapter serving, dynamic checkpoint loading, or RL sampling.
• Familiarity with CUDA graphs, custom kernels, and NVIDIA profiling tools.
• Experience operating model-serving systems under production traffic.

Logistics & Benefits

• Location: Palo Alto, California.
• Compensation: $200,000–$420,000 USD annual base pay, depending on experience and skills, plus equity.
• Benefits: Comprehensive health, dental, and vision insurance; unlimited PTO; and relocation assistance as needed.
• Visa Sponsorship: We sponsor visas and support the process for the right candidate.