← All roles

ML Compiler & System

Genesis· Engineering & Research
Apply on company site
Bay AreaFull-timePosted 5mo ago

About the role

Senior role owning Quadrants end-to-end: API, IR, lowering, optimization passes, numerical primitives, and roadmap for a Python-to-kernel compiler targeting CPU and GPU with Genesis as the primary workload.

Responsibilities

  • API & authoring experience. The surface that physics and numerical-computing engineers write against, and the product’s face to the world. Idiomatic, natural to Python users, never a separate language bolted onto Python. Clean decorators, legible constraints, sharp errors, real escape hatches.
  • IR, lowering & backends. A stable IR that lowers to multiple backends, with room to grow and the door kept open for differentiable simulation.
  • Optimization. Turn naive code into strong execution plans: vectorization, register and shared-memory usage, barriers and atomics, cooperative threading, dead-buffer and redundant-traffic elimination, dropping needless synchronization.
  • Primitives. Cholesky, eigendecomposition, matmul, reductions, solver kernels. Each owned with its correctness, benchmarks, and regression tracking.
  • Watching Genesis and the compiler together. Hunt for performance wins on both sides, in Genesis and in Quadrants itself. Propose API changes that cut duplication, improve readability, and reduce platform-specific special-casing.
  • The project as a product. Benchmark real workloads, track performance over time, catch regressions early. Stay close to users (Genesis first), find friction and missing abstractions, drive adoption, set direction.

Qualifications

  • GPU-focused intuition (memory access patterns, registers, shared memory, barriers, atomics, threading). CPU intuition a plus.
  • Idiomatic Python and API design sensibility. APIs that feel natural to Python users.
  • Experience with compilers: from authoring to kernels (IR, lowering, optimization), balanced with maintainability.

Skills

Required:API designPythonCompiler IR and lowering
Last updated Aug 20, 2026. Interested? Apply on the company site.

About Genesis

A new future is at hand.

We are a new generation of robotics builders, united by a shared mission to push the boundaries of physical AI.

AIGENEGenesisfrontier-labsheadline-ai-europe-100-2026intelligencerobotics
Company profileWebsite

More open roles at Genesis