Lead compiler development focusing on ML compilers for robotics simulation platform. Collaborate with engineers to enhance performance and support for differentiable programming.
Responsibilities
Lead the evolution of our high-performance robotics simulation platform
Design and implement the compute infrastructure and data flow mechanisms to optimize performance for physics simulation and foundation model training
Lead development of our compiler stack, focusing on JIT compilation, LLVM IR, and GPU codegen to minimize compile time and maximize runtime performance
Collaborate with the team to improve the compiler's support for differentiable programming, crucial for training neural networks within simulations
Stay current 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 align compiler enhancements with application needs
Contribute to relevant open-source projects and participate actively in the broader compiler and systems community
Requirements
Strong background in compiler construction, particularly in JIT compilation and LLVM-based code generation
Extensive experience with GPU programming models (e.g., CUDA, Vulkan) and understanding of GPU architecture
Track record as a core contributor to GPU programming infrastructure—such as Torch, JAX, Mojo, Taichi, or Warp
Proven ability to profile and optimize complex systems for performance and scalability
Understanding of automatic differentiation and its application in simulation and machine learning contexts
Excellent communication skills and a collaborative approach to problem-solving
Enthusiasm for contributing to and engaging with open-source communities
Job title
Member of Technical Staff, ML Compiler and Systems
Mobile Developer creating secure login apps using React Native for heylogin. Collaborating with cross - functional teams in a hybrid environment based in Braunschweig, Germany.
Application Developer II managing development of full - stack solutions for a distributor. Collaborating with teams on software application development and maintenance across various technologies.
Technical Staff designing and optimizing distributed training systems for GPU clusters. Aiming to reduce convergence time through efficient coding and infrastructure optimization.
Member of Technical Staff focused on building LM/VLM - powered agents and generative simulation systems. Collaborating with internal teams and driving innovation in robotics and AI applications.
Member of Technical Staff responsible for data pipelines and infrastructure for robotics AI. Collaborating with a team to standardize and unify data processing workflows at scale.
Member of Technical Staff developing end - to - end Vision - Language - Action models for robotics. Collaborating with robotics teams to curate datasets and improve machine learning models.
Member of Technical Staff developing rendering systems for robotics foundation models in Paris. Collaborating with a team to build general - purpose Physical AI.
Member of Technical Staff developing GPU - based simulation pipelines for robotics. Collaborating on essential features to bridge the sim - to - real gap in robotics engineering.
Member of Technical Staff focused on building low - latency inference pipelines for robotics. Designing GPU inference systems and optimizing workloads for efficiency and performance.