MTS, ML Compiler and Systems

Build and evolve a high-performance robotics simulation platform by designing compute infrastructure and data-flow mechanisms that accelerate physics simulation and foundation model training. Lead development of the compiler stack using JIT compilation, LLVM IR, and GPU codegen to reduce compile times and improve runtime performance. Collaborate with simulation and robotics engineers to strengthen differentiable programming support, and stay current with ML compiler techniques across torch, Triton, and JAX while contributing to open source.