R

AI Compiler Engineer

river ai • Texas Metropolitan Area
Visa Sponsorship Relocation
This Job is No Longer Active This position is no longer accepting applications
AI Summary

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 custom silicon, including instruction scheduling, register allocation, and hardware-specific code generation. Collaborate with AI researchers, modelers, performance engineers, and silicon architects to optimize performance and co-design the ISA.

Key Highlights
MLIR/XLA deep learning compiler expertise
Custom silicon backend development
PyTorch internals and graph lowering
Hardware-software co-design

Looking to advance your Development & Programming career with relocation support? Explore Development & Programming Jobs with Relocation Packages that include comprehensive packages to help you move and settle in your new role.

Key Responsibilities
Graph Lowering & Optimization: Design and implement compiler passes to lower PyTorch models into custom hardware, leveraging MLIR dialects and LLVM frameworks.
Custom Backend Development: Develop and maintain the backend toolchain for custom silicon, including instruction scheduling, register allocation, and hardware-specific code generation.
Memory & Loop Transformations: Design sophisticated tiling and fusion strategies to maximize bandwidth utilization and minimize on-chip memory movement.
Kernel Integration: Collaborate with software and hardware teams to integrate high-performance kernels (Triton/CUDA-like) into the automated compiler flow.
Performance Profiling: Identify compilation gaps where the compiler fails to achieve peak hardware performance, and collaborate with the performance team for targeted optimizations.
HW/SW Co-Design: Partner with RTL and Architecture teams to change custom ISA definitions.
Technical Skills Required
MLIR PyTorch C/C++ Computer Architecture

Discover our full range of relocation jobs with comprehensive support packages to help you relocate and settle in your new location.

Benefits & Perks
Generous health, dental, and vision benefits
Unlimited PTO
Relocation support as needed

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, custom training infrastructure, next-generation UIs, and frontier deep learning research.

Who we are

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.

About the Role

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
Job Overview
Posted Aug 12, 2026
Employment Full-time
Experience Mid-Senior level
Location Texas Metropolitan Area
Category Programming
Company river ai
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

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.