Modular

Driver Engineer

USA

Sponsorship not specified$167k-$242kDetected 55 days ago

C++Node.jsCode ReviewGitAPI DevelopmentPyTorchLLMsCommunication

About the role

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

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.
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.
Participate in design discussions, code reviews, and collaborative software development to uphold high engineering standards.
Solid instincts for library and API design - naming, layering, ergonomics.
Team Building Events.

Helpful but not required Experience with asynchronous runtimes, custom memory allocators, or RDMA-based networking (Infiniband, RoCE, EFA, UCX, NIXL, MPI, NVSHMEM/ROCSHMEM/OpenSHMEM).
Experience with zero-copy tensor interoperability - DLPack, the CUDA Array Interface, or similar.
An advanced degree in Computer Science or a related field.

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.
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.
Improve diagnostics and error reporting across our asynchronous execution stack - turning low-level driver failures into actionable messages for kernel and graph authors in production environments.
Hands-on experience with at least one accelerator driver-level API - CUDA Driver API, HIP, Metal, or Vulkan compute - including streams, events, contexts, and module loading.
Familiarity with async allocators and IPC is a plus.
Working understanding of accelerator execution and memory models - stream ordering, host↔device transfers, and pinned memory.

The estimated base salary range for this role to be performed in the US, regardless of the state, is $167,000.00 - $242,000.00 USD.
The estimated base salary range for this role to be performed in Canada, regardless of the province, is $164,000.00 - $237,000.00 CAD.

World-class Benefits.
Premier insurance plans, up to 5% 401k matching, flexible paid time off, and more are available to you!
We offer very strong compensation packages, including stock options.
The total compensation for a candidate will also include annual target bonus, equity, and benefits, with equity making up a significant portion of your total compensation.
Please note that specific benefit packages may vary based on your location.

Contribute to a fully open source project - everything you build will be public and part of our GitHub repo.
A proactive, collaborative mindset, intellectual curiosity, and drive to identify and solve complex technical challenges as part of a high-performing team.
We are a progressive and agile team with some of the industry's best engineering and product leaders.

This listing is sourced directly from Modular's careers page and normalized into a canonical job model.