Software Engineer, Hardware Enablement

About Modular

About the role:

What you will do:

• Implement and validate support for new hardware architectures across the Modular stack, working under the guidance of senior engineers on the team
• Write and optimize Mojo kernels targeting novel accelerator architectures, with a focus on correctness first and performance iteration
• Contribute to cross-team efforts improving portability infrastructure, tooling, and debugging workflows for new target hardware
• Collaborate with hardware vendor engineers to understand target platforms, build integration tests, and triage platform-specific issues
• Develop working knowledge of new hardware platforms — including ISA documentation, memory hierarchies, and vendor toolchains — and share findings with the team through demos and write-ups
• Participate in company events such as on-sites and hackathons, contributing to a collaborative and open engineering culture

What you bring to the table:

• 5+ years of experience in high-performance computing, compiler engineering, or related domains in industry or research
• Familiarity with how AI operators are implemented at a low level (e.g., experience writing or modifying GPU kernels, custom operators, or working with frameworks like PyTorch at the C++ layer)
• Proficiency in C++ and experience working in complex, multi-component software systems
• Hands-on experience with at least one heterogeneous programming model (CUDA, SYCL, OpenCL, or similar), either as a user or contributor
• Some exposure to non-GPU accelerator architectures (DSPs, NPUs, or other hardware accelerators) is a strong plus
• Curiosity and willingness to learn new hardware platforms quickly, comfortable reading architecture manuals and vendor documentation
• A collaborative, team-oriented attitude and alignment with our culture

Helpful, but not required:

• Experience with GPU DSLs/DSELs such as Triton, CUTLASS, or CuTe
• Familiarity with MLIR or LLVM compiler infrastructure
• Experience working directly with hardware vendor teams or on platform bring-up efforts
• Exposure to model serving or inference optimization workflows

What Modular brings to the table:

• Amazing Team. We are a progressive and agile team with some of the industry’s best engineering and product leaders.
• World-class Benefits. In order to attract the best, we need to offer the best. Premier insurance plans, up to 5% 401k matching, flexible paid time off, and more are available to you! Please note that specific benefit packages may vary based on your location.
• Competitive Compensation. We offer very strong compensation packages, including stock options. We want people to be focused on their best work and believe in tailoring compensation plans to meet the needs of our workforce. 
• Team Building Events. We organize regular team onsites and local meetups in Los Altos, CA as well as different cities. Traveling 2-4 times a year is expected for all roles.