R

Senior Systems Engineer (AI Training Infrastructure)

river ai • United States
Visa Sponsorship Relocation
This Job is No Longer Active This position is no longer accepting applications

Build and optimize high-performance, distributed AI training systems at River AI, focusing on scalable GPU kernels, fault-tolerant clusters, and end-to-end performance optimization. Collaborate with researchers to accelerate model development while ensuring reliability and efficiency across thousands of nodes. Requires deep expertise in systems programming, computer architecture, and distributed systems.

Key Highlights
Architect and deploy fault-tolerant distributed systems for AI training and inference across large-scale clusters
Design high-performance GPU kernels and optimize tensor operations, memory, and networking (InfiniBand/RDMA)
Partner directly with research scientists to implement, optimize, and scale experimental model architectures
Key Responsibilities
Design and deploy fault-tolerant distributed systems for AI training and inference workloads across clusters with thousands of nodes
Develop high-performance GPU kernels to maximize tensor operation efficiency, memory throughput, and networking over InfiniBand/RDMA
Profile and optimize systems end-to-end to resolve bottlenecks in hardware, software, data loading pipelines, and collective communication primitives
Collaborate directly with research scientists to implement, optimize, and scale experimental model architectures
Technical Skills Required
C/C++ Distributed Systems Computer Architecture
Benefits & Perks
Comprehensive health, dental, and vision insurance
Unlimited paid time off (PTO)
Relocation assistance as needed
Nice to Have
Experience with modern AI frameworks (e.g., PyTorch, JAX)
Deep familiarity with modern GPU architectures (NVIDIA/AMD) and hardware constraints (HBM, PCIe)
Proven track record of shipping and maintaining high-performance distributed systems or low-level software libraries

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 areWe 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.
Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

Hidden company

Log in to see location

Premium

Senior AI Engineer

Programming Full-time 23s ago

Log in free to see the details

Career Resources & Guides

Latest guides related to this job: Visas, Work Permits & Legal Insights, Relocation Guides.

Browse all articles

Subscribe to our newsletter

New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.