> Markdown version of [/jobs/ext/4606952-software-engineer-distributed-training](https://www.wearedevelopers.com/jobs/ext/4606952-software-engineer-distributed-training). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Software Engineer, Distributed Training

- **Company:** River AI
- **Location:** Palo Alto, CA, United States
- **Salary:** $200,000.0
- **Contract:** Permanent contract
- **Skills:** Application Programming Interfaces (APIs), Artificial Intelligence, Systems Engineering, C++ (Programming Language), Computer Clusters, Nvidia CUDA, Computer Engineering, Distributed Computing Environment, Memory Management, Design of User Interfaces, Human-Computer Interaction, Python (Programming Language), Software Engineering, Reinforcement Learning, Graphics Processing Unit (GPU), Pytorch, Deep Learning, Parallel Computation, Information Technology
- **Published:** November 6, 2026
- **Apply:** https://www.careerbuilder.com/job-details/software-engineer-distributed-training-palo-alto-ca--4bd57f2a-1725-4e8e-b1ac-c558a056192e

## About the Role

Software Engineering, Deep Learning, Editing, Artificial Intelligence (AI), User Interface/Experience (UI/UX), Systems Engineering, Reinforcement Learning, Application Programming Interface (API), GPU (Graphics Processing Unit), Memory Hardware, * 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.

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

 * Experience with reinforcement-learning infrastructure, rollout generation, or asynchronous training.
 * Familiarity with tensor, pipeline, expert, or data parallelism and their performance tradeoffs.
 * Work on LoRA, mixture-of-experts training, distributed optimizers, or activation checkpointing.
 * Experience with NCCL, communication profiling, and overlapping computation with data transfers.
 * Proficiency in C++, Rust, or CUDA, with experience investigating performance below the framework layer.
 * Contributions to training frameworks or a track record of operating large training runs.

## Description

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.

You will own the execution of training workloads, including gradient computation, optimizer updates, rollout coordination, and checkpoint recovery. Working closely with researchers and inference engineers, you will bring new learning methods into production and improve how efficiently models use compute., * Build and optimize distributed training for large dense and mixture-of-experts models, including low-rank adapter training.

 * Improve reinforcement-learning pipelines by coordinating sampling, reward computation, training updates, and weight transfer.
 * Optimize GPU memory use, parallelism, and communication to increase training throughput.
 * Implement reliable checkpointing, resumption, and worker recovery while preserving consistent training state.
 * Validate losses, gradients, and optimizer behavior, and diagnose numerical or distributed execution failures.
 * Partner with researchers to implement new algorithms and make them accessible through the River API.