Modular Logo

Modular

AI Kernel Engineering Manager

Posted One Month Ago
Remote
Hiring Remotely in United States
248K-372K Annually
Mid level
Remote
Hiring Remotely in United States
248K-372K Annually
Mid level
Lead and grow a team of approximately 12 kernel engineers developing high-performance inference kernels across NVIDIA, AMD, and ASIC hardware. Own the MAX kernel roadmap, performance optimization, portable kernel architecture, and delivery. Partner with framework, serving, compiler/runtime, hardware enablement, and Qualcomm teams while advancing AI-assisted development, verification, and engineering practices.
The summary above was generated by AI
About Modular
At Modular, a Qualcomm company, we’re on a mission to revolutionize AI infrastructure by systematically rebuilding the AI software stack from the ground up. Our team, made up of industry leaders and experts, is building cutting-edge, modular infrastructure that simplifies AI development and deployment. By rethinking the complexities of AI systems, we’re empowering everyone to unlock AI’s full potential and tackle some of the world’s most pressing challenges.

If you’re passionate about shaping the future of AI and creating tools that make a real difference in people’s lives, we want you on our team. You can read about our culture and careers to understand how we work and what we value.

About the role:

MAX is Modular's inference tech stack, built to run GenAI models fast across NVIDIA, AMD, Qualcomm, Trainium, TPU and more. The MAX Kernel team is the performance foundation of that stack, developing kernels using Mojo, GEMM, attention (MHA/MLA/MSA), MoE routing and grouped matmul, and the portable tile abstractions that let one kernel lower to many targets.

As the Engineering Manager of the MAX Kernel team, you will lead a team of kernel engineers, own the kernel roadmap and the per-model × per-hardware performance optimization to the best vendor and open-source stacks. You will partner closely with the Framework, Serve, Compiler/Runtime and Hardware Enablement teams, and help grow a kernel engineering community that spans Modular and Qualcomm.

LOCATION: Candidates based in the United States are welcome to apply. You can work in our office in Los Altos, CA or remotely from home. Onboarding for new hires is conducted in-person in our Los Altos, CA office.


What you will do:

  •  Lead, hire, coach and grow a team of ~12 kernel engineers across NVIDIA, AMD and ASIC hardwares; own career development, performance management and team health.
  • Own the MAX kernel roadmap and delivery: match or beat the performance of other open-source kernel libraries on each business essential model × hardware target.
  • Support portable kernel architecture (TileTensor / TensorEngine / TileIO, reusable building blocks such as MegaFFN and the attention family) together with tech leads.
  • Partner cross-functionally with Framework, Serve, Compiler/Runtime, Hardware Enablement and Qualcomm kernel teams on priorities, interfaces and escalations.
  • Advance AI-assisted kernel development (kernel agents, fuzz verification, playbooks) and the kernel engineering community across org boundaries.


What you bring to the table:
 
  •  3+ years as an engineering manager leading GPU kernel, compiler, HPC, or ML performance engineering teams. (minimum requirement)
  • Strong cross-functional communication; able to drive technical decisions with tech leads and communicate status, risks and tradeoffs to engineering leadership and customers.
  • Track record of hiring, retaining and growing senior kernel or performance engineers.
  • Working knowledge of writing and optimizing GPU/accelerator kernels (CUDA, HIP/ROCm, Triton, CUTLASS/CuTe or similar).
  • Deep understanding of profiling, benchmarking, and roofline analysis


Minimum Qualifications:

  • Bachelor's degree in Engineering, Information Systems, Computer Engineering, Electrical Engineering, Computer Science, or related field and 8+ years of Software Engineering or related work experience. 
                          OR 
  • Master's degree in Engineering, Information Systems, Computer Engineering, Electrical Engineering, Computer Science, or related field and 7+ years of Software Engineering or related work experience.
                          OR 
  • PhD in Engineering, Information Systems, Computer Engineering, Electrical Engineering, Computer Science, or related field and 6+ years of Software Engineering or related work experience.
AND
  • 4+ years of work experience with Programming Language such as Python, Mojo, Rust, C++, etc.


Helpful, but not required:
 
  • Experience with Mojo or other kernel and tile-level programming models (Triton, TileLang, CuTe)
  • Multi-target experience beyond NVIDIA: AMD, NPUs/ASICs or edge/on-device.
  • Experience with AI-assisted or agentic kernel development.
  • Open-source contributions to kernel libraries or inference engines.


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 is $255,400.00 - $383,200.00 USD. 

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 your total compensation.
For candidates who fall outside of the listed requirements, we nevertheless encourage you to apply as we may have upcoming openings that are lower/higher level than the ones advertised. 
HQ

Modular Los Altos, California, USA Office

Los Altos, CA, United States

Similar Jobs

Senior level
eCommerce • Information Technology • Internet of Things • Mobile • On-Demand • Retail • Software
Provides senior advisory, pursuit, commercial, and delivery leadership for a Risk & Resilience consulting practice. The role shapes enterprise opportunities, oversees client engagements, governs pricing and scope, advises executives, develops offerings, and connects risk, resilience, cybersecurity, AI, cloud, governance, and compliance capabilities into actionable strategies.
Top Skills: Artificial IntelligenceCloudCybersecurity
Senior level
eCommerce • Information Technology • Internet of Things • Mobile • On-Demand • Retail • Software
Leads Risk & Resilience consulting pursuits and client engagements from opportunity development through delivery oversight. Advises senior enterprise executives on risk, resilience, cybersecurity, AI, cloud, governance, and compliance. Oversees solution shaping, proposals, pricing, SOWs, commercial governance, engagement quality, and delivery risks. Develops repeatable advisory offerings, methodologies, intellectual property, and practice capabilities while mentoring consultants and supporting sales enablement and practice growth.
Top Skills: Artificial IntelligenceCloud ComputingCybersecurity
Senior level
eCommerce • Information Technology • Internet of Things • Mobile • On-Demand • Retail • Software
Leads Risk & Resilience consulting pursuits and client engagements, advising executives on enterprise risk, cybersecurity, cloud, AI, governance, compliance, and operational readiness. Oversees solution shaping, proposals, SOWs, pricing, commercial governance, delivery quality, client relationships, and engagement risks. Develops advisory offerings, methodologies, intellectual property, and practice capabilities while mentoring consultants and supporting business growth.
Top Skills: Artificial IntelligenceCloud ComputingCybersecurity

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account