Member of Technical Staff, Hardware, Compiler Engineer

River AI is on a mission to create personal AI owned and shaped by individuals, and they are seeking exceptional AI compiler engineers to build the software bridge between AI models and custom silicon. The role involves designing compiler passes, developing backend toolchains, and collaborating with hardware and software teams to optimize performance.

Responsibilities:

• Design and implement compiler passes to lower PyTorch models into custom hardware, leveraging MLIR dialects and LLVM frameworks
• Develop and maintain the backend toolchain for our custom silicon, including instruction scheduling, register allocation, and hardware-specific code generation
• Design sophisticated tiling and fusion strategies to maximize bandwidth utilization and minimize on-chip memory movement
• Collaborate with software and hardware teams to integrate high-performance kernels (Triton/CUDA-like) into the automated compiler flow
• Identify "compilation gaps" where the compiler fails to achieve peak hardware performance, and collaborate with the performance team for targeted optimizations to close those gaps
• Partner with the RTL and Architecture teams to change the custom ISA definitions