Apply on Employer Site

Genesis AI · 1 month ago

Staff Software Engineer, ML Compiler and Systems (Bay Area / Paris / Remote)

United States

Full-time

Remote

Lead/Staff

Genesis AI is a company focused on advancing artificial intelligence technologies, and they are seeking a Staff Software Engineer to lead the development of their robotics simulation platform. The role involves designing compute infrastructure, optimizing performance for simulations, and enhancing compiler support for machine learning applications.

Artificial Intelligence (AI)Robotics

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

Qualification

Compiler constructionJIT compilationLLVM-based code generationGPU programming modelsPerformance optimizationAutomatic differentiationCollaborative problem-solvingOpen-source engagementCommunication