River AI Verified

AI Infrastructure, Model Serving

Software Engineer, Inference Systems

Name: RiseMe
Availability: InStock

Palo Alto, CAOnsiteFull Time$200,000–$420,000 /yrPosted 1 month ago

Is this role right for you?

Upload your resume and get a skill-by-skill breakdown — see exactly where you match, where you're close, and what to highlight. Not a mystery percentage.

Get a tailored resume highlighting what this role needs.

Required skills

inference systems
inference engines
serving runtime
request scheduling
continuous batching
KV-cache management
distributed model execution
checkpoint loading
customer-facing inference
RL sampling

Nice to have

SGLang
vLLM
TensorRT-LLM
speculative decoding
disaggregated prefill
expert parallelism

Role summary

River AI is seeking exceptional Inference Systems Engineers to build and optimize the serving infrastructure for large AI models via the River API. This role involves delivering fast, reliable, and cost-efficient inference by optimizing GPU compute and memory usage. You will own the serving runtime, focusing on request scheduling, batching, caching, distributed execution, and model loading. Collaboration with GPU kernel engineers, researchers, and infrastructure engineers is key to bringing new models into production and enhancing their performance. Success will be measured by latency, throughput, reliability, and cost, ensuring numerical correctness.

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.

Ready to apply?

You'll be redirected to River AI's application page.