Driver Engineer
Modular · CA
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, synch...
Job description
About the role 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. Your work directly affects how reliably and efficiently MAX and Mojo run across NVIDIA, AMD, Apple Silicon, and emerging accelerators, and how cleanly kernel and graph authors can target them from Mojo and Python. This is the team that makes "it just works on a new accelerator" actually true. LOCATION: Candidates based in the US or Canada are welcome to apply. You can work in our office in Los Altos, CA or remotely from home. What you will do: What you bring to the table - Implement and extend core driver abstractions — Device, Context, Queue, Memory, Function (kernel representation and launch) — across diverse hardware backends, including bindings that expose vendor-specific functionality where it matters. - Contribute to multi-accelerator and multi-node communication and collectives primitives that underpin large-model inference, using technologies like NVLink, RDMA (Infiniband, RoCE, EFA), an...