# Software Engineer, Distributed Training at River AI

- Company: River AI
- What the company does: Develop frontier language models and agents. Training, reinforcement learning, and inference in one system, on River Cloud or your own GPU cluster. Backed by General Catalyst.
- Company website: https://river.ai/
- Type: Startups (AI role)
- Level: Mid level
- Location: Palo Alto, CA
- Work setup: On-site
- Pay: $200K to $420K base salary per year (USD)
- Posted: 2026-09-11
- Apply by: 2026-10-26
- Apply: https://job-boards.greenhouse.io/riverai/jobs/4402494009
- Page: https://www.1752.vc/careers/jobs/river-ai-software-engineer-distributed-training/

## About the role

We are looking for exceptional systems engineers to build the distributed training engines behind the River API. Your goal is to make fine-tuning and reinforcement learning fast, numerically correct, and reliable across large GPU clusters.

## What they're looking for

- Bachelor’s degree in Computer Science, Computer Engineering, or equivalent practical industry experience
- Hands-on experience building or substantially improving distributed model-training systems
- Strong proficiency in Python and a modern deep-learning framework, such as PyTorch or JAX
- Solid understanding of backpropagation, optimizers, mixed-precision training, and GPU memory management
- Strong debugging skills across concurrent execution, collective communication, and distributed failure recovery
- A highly collaborative mindset and a bias for action to push boundaries across the stack

Tags: River API