ML Compiler & System

Genesis AI
Paris, London
Workplace: HybridFull timeFunction: Software EngineeringSkills: ["Communication","Collaboration","Problem-solving"]

Lead the evolution of a high-performance robotics simulation platform by designing compute infrastructure and data-flow mechanisms for physics simulation and foundation model training. Develop a compiler stack centered on JIT compilation, LLVM IR, and GPU codegen to reduce compile times while improving runtime performance. Collaborate with simulation and robotics engineers on differentiable programming support, stay current with ML compiler research, and contribute to relevant open-source projects.

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

Lead the evolution of a high-performance robotics simulation platform by designing compute infrastructure and data-flow mechanisms for physics simulation and foundation model training. Develop a compiler stack centered on JIT compilation, LLVM IR, and GPU codegen to reduce compile times while improving runtime performance. Collaborate with simulation and robotics engineers on differentiable programming support, stay current with ML compiler research, and contribute to relevant open-source projects.
Location: Paris, London
Workplace: Hybrid
Employment Type: Full time
Job Function: Software Engineering

Key Responsibilities

  • •Lead the evolution of a high-performance robotics simulation platform.
  • •Design compute infrastructure and data-flow mechanisms to optimize performance for physics simulation and foundation model training.
  • •Develop the 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 compiler support for differentiable programming for training neural networks within simulations.
  • •Stay current on ML compiler techniques and partner with simulation and robotics engineers to align compiler enhancements with application needs.

Key Requirements

  • •Strong background in compiler construction, especially JIT compilation and LLVM-based code generation.
  • •Extensive experience with GPU programming models such as CUDA and Vulkan, with knowledge of GPU architecture.
  • •Track record contributing to GPU programming infrastructure (e.g., Torch, JAX, Mojo, Taichi, or Warp).
  • •Proven ability to profile and optimize complex systems for performance and scalability.
  • •Understanding of automatic differentiation and its use in simulation and machine learning contexts.
Experience:RoboticsMachine learningOpen source
Skills:CommunicationCollaborationProblem-solving
Tech Stack:LLVM IRLLVMJIT compilationGPU 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