Lead Python ML Compiler & GPU Systems Engineer

Genesis

Northern (KY)

Hybrid

USD 130,000 - 190,000

Full time

14 days+
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Job summary

Genesis is seeking a senior role to own Quadrants' end-to-end API, IR, and optimization passes. You will shape the surface for Pythonic numerical computing, lower computations to multiple backends, and push performance across GPU backends.

The role emphasizes readability, maintainability, and strong engineering discipline while enabling differentiable simulation. You will drive practical kernel generation, propose API changes, and mentor others, with Genesis first as a priority user and

Qualifications

  • Deep knowledge of GPU architectures and memory models.
  • Strong Python programming experience with idiomatic API design.
  • Understanding of compilers, IR, lowering, and optimizations.

Responsibilities

  • Own API, IR and lowering, optimization passes, numerical primitives, and roadmap.
  • Build stable IR that lowers to multiple backends and supports differentiable simulation.
  • Transform naive code into efficient execution plans using vectorization, memory optimization, and threading.
  • Develop primitives like Cholesky, eigendecomposition, matmul, and reductions with benchmarks and regression tracking.
  • Collaborate with Genesis on performance wins and API evolution, reducing duplication and improving usability.
  • Lead a small team over time and engage with users to shape the product direction.

Skills

GPU knowledge
Python & API design
Compiler theory

Job description

Genesis is seeking a senior role to own Quadrants' end-to-end API, IR, and optimization passes. You will shape the surface for Pythonic numerical computing, lower computations to multiple backends, and push performance across GPU backends.

The role emphasizes readability, maintainability, and strong engineering discipline while enabling differentiable simulation. You will drive practical kernel generation, propose API changes, and mentor others, with Genesis first as a priority user and

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Lead Python Compiler Engineer for Physics & GPUs
Lead Python Compiler Engineer for Physics & GPUs

Genesis AI • Northern (KY)

Hybrid
USD 120,000 - 180,000
ML Compiler & System Engineering & Research Bay Area
ML Compiler & System Engineering & Research Bay Area

Genesis • Northern (KY)

Hybrid
USD 130,000 - 190,000
Lead Core Physics Engineer for Next-Gen Robotics Simulation
Lead Core Physics Engineer for Next-Gen Robotics Simulation

Genesis AI • San Francisco (CA)

On-site
USD 150,000 - 230,000
Quadrants: Compiler Lead
Quadrants: Compiler Lead

Genesis AI • Northern (KY)

On-site
USD 120,000 - 180,000
Open-Source Core Simulation Platform Engineer
Open-Source Core Simulation Platform Engineer

Genesis AI • San Francisco (CA)

On-site
USD 180,000 - 240,000
Senior ML Training Systems Engineer - Distributed CUDA
Senior ML Training Systems Engineer - Distributed CUDA

Genesis AI • San Francisco (CA)

On-site
USD 180,000 - 260,000
Senior ML Infra Engineer for Distributed GPU Training
Senior ML Infra Engineer for Distributed GPU Training

Genesis Molecular AI • City of Utica (NY)

On-site
USD 150,000 - 190,000
Competitive compensation with salary +
Genesis-World: Core Simulation Engine Engineer
Genesis-World: Core Simulation Engine Engineer

Genesis AI • San Francisco (CA)

On-site
USD 180,000 - 240,000
Machine Learning Infrastructure Engineer
Machine Learning Infrastructure Engineer

Genesis Molecular AI • City of Utica (NY)

On-site
USD 150,000 - 190,000
Competitive compensation with salary +
Open-Source Compiler Lead for Physics Simulations
Open-Source Compiler Lead for Physics Simulations

United States Digital Space LLC • Paris (TX)

On-site
USD 120,000 - 180,000