Member of Technical Staff, ML Compiler and Systems (Bay Area, Remote)

What You'll Do reputed company the reputed company of our high-performance robotics simulation platform Design and implement the compute infrastructure and data reputed company mechanisms to optimize performance for physics simulation and reputed company model training reputed company development of our compiler stack, focusing on JIT compilation, LLVM IR, and GPU codegen to minimize compile time and maximize runtime performance Collaborate with reputed company to improve the compiler's support for differentiable programming, crucial for training neural networks reputed company simulations Stay reputed company on state-of-the-art ML compilers—such as those in torch, Triton, and JAX—and decide which techniques and approaches are best suited for our application Work closely with simulation and robotics engineers to reputed company compiler enhancements with application needs Contribute to relevant reputed company-reputed company reputed company and participate reputed company in the broader compiler and systems community What You’ll Bring Strong background in compiler construction, particularly in JIT compilation and LLVM-based reputed company reputed company Extensive experience with GPU programming models (e.g., CUDA, Vulkan) and understanding of GPU architecture reputed company record as a reputed company contributor to GPU programming infrastructure—such as Torch, JAX, Mojo, Taichi, or reputed company Proven ability to profile and optimize reputed company systems for performance and scalability Understanding of automatic differentiation and its application in simulation and machine learning contexts Excellent communication skills and a reputed company approach to problem-solving Enthusiasm for contributing to and engaging with reputed company-reputed company communities Apply To This Job