Genesis Logo

Genesis

ML Compiler & System

Posted 15 Days Ago
Be an Early Applicant
In-Office or Remote
Hiring Remotely in CA
Senior level
In-Office or Remote
Hiring Remotely in CA
Senior level
Lead development of a high-performance robotics simulation platform and its compiler stack. Design compute infrastructure and data flow for physics simulation and model training, optimize GPU codegen and JIT compilation, improve differentiable programming support, and contribute to open-source ML compiler projects while collaborating with simulation and robotics engineers.
The summary above was generated by AI
What Quadrants is

Quadrants is an open-source translation layer that turns ordinary Python into optimized machine code at runtime, across CPU (Arm64, x86) and GPU (AMDGPU, CUDA, Metal, Vulkan) backends. CUDA is the primary target, with x86 the strong runner-up, especially for real-time simulation. The others drive adoption and support fast prototyping for research.

The ambition is explicit: build the best “Python compiler” in the world for physics and performance-critical numerical computing. We are fully committed to it, and we intend to beat leading alternatives like Nvidia Warp, Numba, or Triton.

It’s a standalone project with its own roadmap, built in the open. Genesis, our physics simulator, is its first and top-priority user, the source of its hardest workloads and its benchmarks. But Quadrants is general-purpose infrastructure for anyone writing performance-critical numerical Python, not an internal tool that happens to be public.

The bet: high-performance code should feel natural to write. Engineers write plain Python in a NumPy or PyTorch style, and Quadrants makes it fast.

The role

You own Quadrants end to end: the API, the IR and lowering, the optimization passes, the numerical primitives, and the roadmap.

Our ultimate objective: engineers write for readability and maintainability, never for performance. They write the version of an algorithm they are most comfortable with, and Quadrants rebuilds it into kernels that run as if an expert had hand-tuned them. Performance is the compiler’s job, not theirs. Getting there is a frontier problem, part research and part engineering, and we intend to define the state of the art on it, relentlessly pushing the limit further.

It calls for two things at once: the low-level GPU and compiler skill to make a single kernel optimal, and the ability to keep the whole system coherent as it grows.

The core tradeoff is settled: usability comes first. When usability and performance collide, usability wins, unless the cost is severe. We will limit what the system supports to go fast (static shapes and fixed-size allocation baked into kernels, say), but never the convenience of the API itself. Experts who want maximum performance and full control over the generated code get lower-level escape hatches, down to inline assembly, C, or PTX.

What you’ll own
  • API & authoring experience. The surface that physics and numerical-computing engineers write against, and, as an open-source library, 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. Today the computation graph is expressed through Quadrants’ own constructs. Recovering it automatically from ordinary Python is hard, and a long-term goal rather than today’s reality.

  • 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. Move low-level optimization off users and into the platform.

  • 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. Less to maintain, more that simply works.

  • 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.

Who you are

You reason about whole systems and have taste for where complexity should live. You measure before you optimize, and you reach for memory traffic and launch overhead before FLOPs. You have opinions about API design. You write tests and benchmarks as you go, not afterward.

The ground that judgment stands on:

  • GPU. What optimal looks like and why: access patterns, registers, shared memory, barriers, atomics, threading. CPU intuition a plus.

  • Python & API taste. Idiomatic Python. A feel for what’s natural to Python users. APIs that free users rather than constrain them.

  • Compilers. The path from authoring to kernels (IR, lowering, optimization), weighed against maintainability.

And how you work: proactive and largely self-directed. You fetch context across the company instead of waiting for it. You’re comfortable talking to the engineers who use what you build, and at ease designing in the open for users you’ll never meet. Ready to grow into leading a small team (a couple of people, near term).

What great looks like

People adopt Quadrants from outside Genesis. Engineers write clearer code with less boilerplate. Kernels keep getting faster. Performance is tracked, not guessed. The architecture gets easier to extend, not harder.

What we’re uncompromising about

How you work is up to you. What matters is that your ownership grows over time: blind spots shrink, the parts you can vouch for expand, technical debt goes down, the core stays sound.

That takes treating understanding as real work, not overhead. You read and challenge the codebase instead of just stacking features on it. You take nothing you did not write for granted. You watch the competition and what’s coming next, and you bring it in and ship it.

Use AI agents if they help. They are productivity tools, not code owners. You are.

Similar Jobs

2 Hours Ago
Easy Apply
Remote or Hybrid
Canada
Easy Apply
Senior level
Senior level
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Design and build Samsara’s cloud governance platform, including IAM guardrails, security controls, compliance automation, and software supply-chain security tooling across AWS and GCP. Develop reliable distributed systems, secure CI/CD patterns, and SAST/DAST capabilities. Contribute to technical roadmaps, collaborate across infrastructure, security, compliance, and engineering teams, and communicate design tradeoffs while mentoring teammates through reviews and pairing.
Top Skills: AWSCi/CdCloud InfrastructureDastDistributed SystemsGCPGoIamPythonSastTerraform
3 Hours Ago
Easy Apply
Remote
Canada
Easy Apply
Senior level
Senior level
Big Data • Fintech • Mobile • Payments • Financial Services
Lead frontend engineering for Affirm’s consumer web experiences using React and TypeScript. Own quarterly team goals, deliver major product features, modernize web architecture and design systems, and develop tools such as codemods and AI agent skills. Collaborate with product, design, analytics, and engineering teams; establish quality standards; monitor availability and metrics; support on-call operations; resolve technical issues; and mentor engineers.
Top Skills: FigmaJavaScriptReactTypescriptVue
3 Hours Ago
Easy Apply
Remote
Canada
Easy Apply
Expert/Leader
Expert/Leader
Big Data • Fintech • Mobile • Payments • Financial Services
Leads Affirm’s global customer operations vendor strategy and BPO network, owning governance, performance, compliance, resiliency, cost optimization, AI-enabled innovation, and customer experience. Manages Vendor Managers and international stakeholders while overseeing vendor selection, migrations, escalations, operational transformations, analytics, workforce planning, and recovery efforts. Builds scalable governance and high-performing, customer-centric vendor organizations across global markets.
Top Skills: Artificial Intelligence (Ai)AutomationBpo

What you need to know about the Ottawa Tech Scene

The capital city of Canada and the nation's fourth-largest urban area, Ottawa has proven a rapidly growing global tech hub. With over 1,800 tech companies, many of which are leaders in their sectors, the city's tech talent now makes up more than 13 percent of its total workforce. This growth is driven not only by the big players like UL Solutions and Dropbox, but also by a thriving startup ecosystem, as new businesses emerge to follow in the footsteps of those that came before them.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account