Software Engineer, GPU Kernels

We are looking for exceptional GPU kernel engineers to build the compute primitives behind River’s training and inference infrastructure. Your goal is to make large models faster to train and more efficient to serve. You will own performance-critical operations, including attention, matrix multiplication, mixture-of-experts execution, and low-precision computation. Working closely with researchers and systems…

River AI Inc. has 24 open roles right now. Most are in Engineering, Data & Research and Operations. The median advertised salary across its roles with a published range is $310K. See all its roles