Modular · 1 day ago
GenAI Optimization Technical Lead
United States
Full-time
Remote
Senior Level, Lead/Staff
$234K/yr - $286K/yr
6+ years expModular is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack from the ground up. They are seeking a GenAI Optimization Technical Lead to guide a team in optimizing generative AI use cases and drive the technical roadmap for their AI platform, MAX.
AI InfrastructureArtificial Intelligence (AI)Generative AIMachine LearningSoftware
Responsibilities
Serve as Technical Leader for a team of cross-functional engineers optimizing Generative AI use cases
Mentor and contribute to the growth of other engineers
Define and drive the technical roadmap for performance architecture for the MAX Serve SDK
Define, plan, and technically lead major cross-functional projects spanning multiple teams and domains to drive core performance and scalability wins
Serve as a primary technical liaison, collaborating with subject matter experts across Modular to align on performance features and stack integration
Contribute to the MAX tech stack across multiple languages, including Mojo, Python, and C++
Monitor latest research channels and identify potential opportunities for the MAX framework
Qualification
Required
In-depth knowledge of the Python programming language
6+ years of working experience in Machine Learning, Deep Learning, or Generative AI
Proven track record of designing and delivering high-performance, scalable systems or frameworks for AI inference and serving
Experience leading technical teams to execute in challenging problem spaces
Experience implementing framework-level performance optimizations for Generative AI use cases
Experience profiling and reducing latency in GenAI applications
Deep interest in machine learning technologies and use cases
Creativity and curiosity for solving complex problems, a team-oriented attitude that enables you to work well with others, and alignment with our culture
CUDA/ROCM/Accelerator Programming and Optimization experience
Preferred
Experience using Machine Learning frameworks like PyTorch, Tensorflow, etc
Experience with LLVM/MLIR/Compilers
Experience working with distributed/parallel programming models and an understanding of parallel hardware
Benefits
Premier insurance plans
Up to 5% 401k matching
Flexible paid time off
Annual target bonus
Equity
Benefits
Company
Modular
Modular provides AI infrastructure for deployment, serving, and programming GPUs.
51-200 employees
H1B Sponsorship
Modular has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (6)
2023 (8)
2022 (4)
Funding
Current Stage
Growth StageTotal Funding
$380MKey Investors
US Innovative Technology FundGeneral CatalystGoogle Ventures
2025-09-24Series C· $250M
2023-08-24Series B· $100M
2022-06-30Seed· $30M
Leadership Team
Chris Lattner
CEO + Co-Founder
Tim Davis
Co-Founder & President
Recent News
Greylock
2025-12-29
Crunchbase News
2025-10-15
Company data provided by crunchbase