ML Compiler and Systems
Core
Lead the evolution of a high-performance robotics simulation platform by designing compute infrastructure and a compiler stack for physics simulation and foundation model training.
Role type
Senior IC ML Compiler and Systems Engineer
Builds
High-performance robotics simulation platform, compiler stack (JIT, LLVM IR, GPU codegen)
Domain
Robotics simulation, Machine Learning, Systems programming
Deliverable
production ML models | infrastructure
Required skills
JIT compilation, LLVM-based code generation, GPU programming models (CUDA, Vulkan), GPU architecture, profiling and optimization, automatic differentiation, open-source contribution
Preferred skills
Experience with ML compilers (Torch, Triton, JAX), differentiable programming
Technologies
LLVM, CUDA, Vulkan, Torch, JAX, Mojo, Taichi, Warp
Responsibilities
Design and implement compute infrastructure and data flow mechanisms; Lead development of compiler stack focusing on JIT, LLVM IR, and GPU codegen; Collaborate on improving compiler support for differentiable programming; Stay current on state-of-the-art ML compilers; Work with simulation and robotics engineers to align compiler enhancements; Contribute to relevant open-source projects
Seniority
Senior, hands-on IC
