ML Compiler & System

Genesis AI
United States
Workplace: OnsiteFull timeFunction: Software EngineeringSkills: ["Communication","Collaborative problem-solving","Open-source engagement"]

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.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Genesis AI
Genesis AI
4 months ago

ML Compiler & System

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 6 hours agoStatus: Live

Job Summary

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.
Location: United States
Workplace: Onsite
Employment Type: Full time
Job Function: Software Engineering
Seniority: Mid level

Key Responsibilities

  • •Lead the evolution of a high-performance robotics simulation platform.
  • •Design and implement compute infrastructure and data-flow mechanisms to optimize physics simulation and foundation model training performance.
  • •Lead development of the compiler stack, using JIT compilation, LLVM IR, and GPU codegen to minimize compile time and maximize runtime performance.
  • •Improve compiler support for differentiable programming to enable neural network training within simulations.
  • •Collaborate with simulation and robotics engineers and contribute to relevant open-source compiler and systems projects.

Key Requirements

  • •Strong background in compiler construction, particularly 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.
Experience:Robotics simulationMachine learningCompilersGPU programming
Skills:CommunicationCollaborative problem-solvingOpen-source engagement
Tech Stack:JIT compilationLLVM IRGPU codegenCUDAVulkanTorchTritonJAXMojoTaichiWarp

Company Brief

Genesis AI
Genesis AI is a global physical AI research lab and full‑stack robotics company building a universal robotics foundation model and horizontal platform to enable general‑purpose robots and scale automation of physical labor.
Industry: AI & Machine Learning
Company Size: Small (11 to 50 employees)
Growth: Early Stage Startup
Funding: Seed
Headquarters: Paris, France
Founded: 2024
WebsiteLinkedIn