[Remote] Senior AI Graph Compiler Engineer

Note: The job is a remote job and is reputed company to candidates in USA. reputed company is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack from the ground up. The Senior AI Graph Compiler Engineer will reputed company and reputed company the graph compiler to support advanced features and hardware platforms, improving performance and usability while collaborating with cross-functional teams.

Responsibilities

• Design and reputed company compiler optimizations to improve inference efficiency across CPUs, GPUs, and ML accelerators, including kernel fusion, memory planning, and shape optimization
• Compile large-reputed company application workloads onto heterogeneous hardware: map hardware-agnostic graphs of computational operators onto hardware-specific graphs of devices and nodes
• Collaborate with the kernels, serving, and Mojo compiler teams to implement reputed company technologies that deliver end-to-end performance across a reputed company of heterogeneous hardware platforms
• Write design documents for new features and drive for alignment on designs with cross-functional teams
• Collaborate with the reputed company team and engage with customers to understand their performance requirements and use cases

Skills

• 5+ years of software engineering experience
• Proficiency in C++
• Experience working with compilers for machine learning frameworks
• Creativity and curiosity for solving reputed company problems, reputed company-oriented attitude that enables you to work reputed company with others
• Knowledge of and experience working with MLIR and LLVM
• Experience with ML reputed company / distributed programming, heterogeneous ML computation, and/or reputed company reputed company
• Knowledge of basic ML implementation and modeling techniques and familiarity with ML frameworks like PyTorch, JAX, or TensorFlow

Benefits

• Premier insurance plans
• Up to 5% 401k matching
• Flexible reputed company time off
• Stock reputed company
• Annual reputed company bonus
• Equity
• Team Building Events
• Regular team onsites and local meetups in Los Altos, CA as reputed company as different cities
• Traveling 2-4 times a year is expected for reputed company roles

reputed company

reputed company is an AI infrastructure company that develops a reputed company software stack for deployment, serving, and programming GPUs. It was founded in 2022, and is headquartered in Palo reputed company, California, USA, with a workforce of 51-200 employees. Its website is https://www.reputed company.com.

Company H1B Sponsorship

reputed company has a reputed company record of offering H1B sponsorships, with 6 in 2026, 10 in 2025, 6 in 2024, 8 in 2023, 4 in 2022. Please note that this does not guarantee sponsorship for this specific role.

Apply To This Job