Mojo Libraries Engineer
Modular · USA ·
- Work mode
- Remote
- Experience
- 2+ years
- Visa sponsorship
- Yes
- Company size
- 501-1000
- Salary
- USD 167,000 – 286,000 / year
About the role:
ML developers today face significant friction in taking trained models into deployment. They work in a highly fragmented space, with incomplete and patchwork solutions that require significant performance tuning and non-generalizable, model-specific enhancements. At Modular, we are building the next generation AI platform that will radically improve the way developers build and deploy AI models.
As part of our mission to build AI's unified compute layer, we are expanding the Modular software stack to a variety of new and exciting hardware platforms.
We are looking for a motivated engineer to join the Hardware Enablement team at Modular. In this role you will work across the Modular software stack, from Mojo kernels and the graph compiler to MAX model serving to bring up and optimize support for new accelerator platforms. You'll collaborate closely with internal teams and external hardware partners, and you'll develop deep expertise in novel architectures while contributing to our portability story.
LOCATION: Candidates based in the US, Canada are welcome to apply. To support growth and collaboration, those in earlier career stages work in a hybrid capacity at our Los Altos, CA office (minimum 3 days per week on-site) with relocation assistance provided for out-of-state candidates based in the US. Senior members have both in office or remote flexibility. Onboarding for new hires is conducted in Los Altos, CA.
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:
- 2+ years of experience in high-performance computing, compiler engineering, or related domains in industry or research
- 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:
- 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)
- 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. Your benefits package may include comprehensive healthcare coverage, retirement and savings programs, employee stock purchase opportunities, paid time off, wellbeing resources, family support programs, and learning and development opportunities. Please note that specific benefit packages may vary based on your location, you can read more about benefits offered by Qualcomm here.
- Competitive Compensation. We offer very strong compensation packages, including RSU grants. 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.
Working at Modular will enable you to grow quickly as you work alongside incredibly motivated and talented people who have high standards, possess a growth mindset, and a purpose to truly change the world.
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,400.00–$258,600.00 CAD.
The salary for the successful applicant will depend on a variety of permissible, non-discriminatory job-related factors, which include but are not limited to education, training, work experience, business needs, or market demands. This range may be modified in the future. The total compensation for a candidate will also include annual target bonus, equity, and benefits, with equity making up a significant portion of total compensation.
For candidates who fall outside of the listed requirements, we nevertheless encourage you to apply, as we may have upcoming openings at lower or higher levels than the one advertised.