ML Compiler & System at Genesis | GB | Rezi

ML Compiler & System at Genesis

ML Compiler & System

Genesis · GB

2 weeks ago

ML Compiler & System

Genesis · GB

16 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this ML Compiler & System role.

Rezi rewrites your resume against Genesis's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the ML Compiler & System posting at Genesis — free, in seconds.

About the Role

You will own Quadrants end-to-end, including the API, IR and lowering, optimization passes, numerical primitives, and the roadmap. The ultimate objective is for engineers to write for readability and maintainability, with Quadrants rebuilding their code into highly optimized kernels. This role requires both low-level GPU and compiler skills, and the ability to maintain system coherence.

Responsibilities

  • Own the API and authoring experience, ensuring it is idiomatic and natural for Python users.
  • Manage the IR, lowering, and backends, supporting multiple targets and future growth.
  • Implement optimizations to transform naive code into efficient execution plans.
  • Develop and maintain numerical primitives, including correctness and performance tracking.
  • Collaborate with Genesis and the compiler team to identify performance wins and propose API improvements.
  • Manage the project as a product, including benchmarking, performance tracking, and user engagement.
  • Drive adoption and set the project's direction.

Requirements

  • Reason about whole systems and understand where complexity should reside.
  • Prioritize measurement before optimization, focusing on memory traffic and launch overhead.
  • Have strong opinions about API design.
  • Write tests and benchmarks proactively.
  • Understand optimal GPU performance, including access patterns, registers, shared memory, barriers, atomics, and threading.
  • Possess CPU intuition as a plus.
  • Demonstrate idiomatic Python and an understanding of what is natural for Python users.
  • Design APIs that empower users rather than restrict them.
  • Understand the compiler path from authoring to kernels (IR, lowering, optimization) and its impact on maintainability.
  • Be proactive and largely self-directed.
  • Fetch context across the company independently.
  • Be comfortable interacting with engineers who use the built software.
  • Be at ease designing in the open for unknown users.
  • Be ready to grow into leading a small team.
  • Treat understanding as real work, not overhead.
  • Read and challenge the codebase.
  • Monitor the competition and emerging trends.

Skills

  • GPU programming
  • Compiler design
  • API design
  • Python development
  • Performance optimization
  • Benchmarking
  • Testing
  • System design

Location

  • Remote

Work Type

  • Full-time

Experience Level

  • Senior
  • Lead

About the Company

  • Quadrants is an open-source translation layer that optimizes Python into machine code for CPU and GPU backends.
  • The company's ambition is to build the world's best Python compiler for physics and performance-critical numerical computing.
  • Quadrants is a standalone project with its own roadmap, built in the open.
  • Genesis, a physics simulator, is the primary user and benchmark source for Quadrants.
  • The core philosophy is that high-performance code should be natural to write in plain Python.
  • The company prioritizes usability, ensuring the API remains convenient even when performance is a concern.
  • Experts have access to lower-level escape hatches for maximum performance and control.