River
Member of Technical Staff, Software Engineering
Palo Alto, CA · Staff+
H1B sponsorship available$200k-$420kDetected 25 days ago
C++Distributed SystemsDeep LearningPyTorchLogisticsResearchCommunication
About the role
- Your goal is to make training at River fast, reliable, and massively scalable.
- You will take ownership of our core infrastructure stack
- from writing custom GPU kernels to managing clusters of thousands of nodes, ensuring our researchers can focus on science rather than system bottlenecks.
Responsibilities
- Design high-performance kernels to maximize tensor operation efficiency, memory throughput, and networking over InfiniBand/RDMA.
- Partner directly with research scientists to rapidly implement, optimize, and scale experimental model architectures.
Requirements
- Bachelor's degree in Computer Science, Computer Engineering, or equivalent practical industry experience.
- Deep expertise in systems-level languages (C, C++, or Rust) with a track record of writing performant, maintainable code.
Nice to have
- (We encourage you to apply even if you don't meet all of these)
- Hands-on experience with modern AI frameworks (e.g., PyTorch, JAX) and tooling for large-scale model training.
- Deep familiarity with modern GPU architectures (NVIDIA/AMD) and hardware constraints (HBM bandwidth, PCIe limits).
- A proven track record of shipping and maintaining high-performance distributed systems or low-level software libraries.
- and relocation assistance as needed.
- Visa Sponsorship: We sponsor visas and are committed to supporting the process for the right candidate.
Compensation
- Depending on experience and skills the expected base pay is $200,000 - $420,000 USD per year.
Benefits
- Comprehensive health, dental, and vision insurance; unlimited PTO; and relocation assistance as needed.
Company info
- 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.
- At River, our mission is to create personal AI owned and shaped by each individual.
- personal hardware for local inference, bespoke training infrastructure, next-generation UIs, and frontier deep learning research.
- We are looking for exceptional systems engineers to build the high-performance engines that train our models. Your goal is to make training at River fast, reliable, and massively scalable.
- We are looking for exceptional systems engineers to build the high-performance engines that train our models.
Visa & Work Authorization
- We sponsor visas and are committed to supporting the process for the right candidate.
This listing is sourced directly from River's careers page and normalized into a canonical job model.