🛠 Tech Stack

💼 About This Role

You'll lead the evolution of our high-performance robotics simulation platform, focusing on JIT compilation and GPU codegen. You'll collaborate with simulation and robotics engineers to align compiler enhancements with application needs, and contribute to open-source projects like Torch and JAX.

🎯 What You'll Do

• Design compute infrastructure for physics simulation and model training
• Develop LLVM-based JIT compiler with GPU codegen
• Improve compiler support for differentiable programming
• Profile and optimize complex systems for performance

📋 Requirements

• 3+ years in compiler construction with JIT and LLVM
• Expertise in GPU programming (CUDA, Vulkan)
• Core contributor to GPU infrastructure like Torch or JAX
• Understanding of automatic differentiation in simulation/ML

✨ Nice to Have

• Experience with robotics simulation platforms
• Open-source contributions in compiler or systems communities