Design and evolve core APIs and developer experience for Modular’s MAX AI framework, focusing on inference and training systems.
Locations (3)
Los Altos, California, United States
United States
Canada
Compensation
The estimated base salary range for this role to be performed in the US, regardless of the state, is $180,000–$324,000 USD. The estimated base salary range for this role to be performed in Canada, regardless of the province, is $172,400.00–$258,700.00 CAD.
Responsibilities
Influence the design of the API surface for MAX
Design inference APIs that support real-world serving needs
Design training APIs that scale from single device to distributed execution
Create a coherent programming model across Python and Mojo-adjacent surfaces
As a Driver Engineer, you'll work on the layer that sits between Modular's compiler and runtime stack and the silicon — the devices, contexts, queues, synchronization primitives, memory allocators, networking, and collectives that every kernel and graph execution depends on.
Locations (2)
Los Altos, California, United States
Canada
Compensation
The estimated base salary range for this role to be performed in the US, regardless of the state, is $148,000.00–$270,000.00 USD. The estimated base salary range for this role to be performed in Canada, regardless of the province, is $172,000.00–$258,600.00 CAD.
Responsibilities
Implement and extend core driver abstractions — Device, Context, Queue, Memory, Function (kernel representation and launch) — across diverse hardware backends.
Contribute to multi-accelerator and multi-node communication and collectives primitives that underpin large-model inference using technologies like NVLink, RDMA (Infiniband, RoCE, EFA), and sockets.
Improve diagnostics and error reporting across our asynchronous execution stack.
Partner with peer teams — Kernels, Graph Compiler/Runtime, and Serving to build and refine the surfaces they consume in the Mojo standard library and MAX framework.
S
Maintenance Operations Associate (PT)
San Antonio Spurs Sports & Entertainment · Posted Nov 4, 2026
Mojo Libraries EngineerModular · Los Altos, US+2 more