Modular · 6 hours ago
GenAI Optimization Intern
Los Altos, CA
Internship
Hybrid
Intern
$47/hr - $65/hrModular is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack. As a GenAI Optimization Intern, you will work on performance optimization for advanced AI models, contributing to the open-source MAX platform and gaining hands-on experience in systems programming and profiling tools.
AI InfrastructureArtificial Intelligence (AI)Generative AIMachine LearningSoftware
Responsibilities
Tackle real performance problems in state-of-the-art GenAI models like DeepSeek, Llama, and Qwen
Execute on a critical project related to performance optimization across the Python/Mojo boundary
Work from GPU kernels up through compiler, framework, and multiprocessing optimizations
Profile bottlenecks in multi-billion parameter models
Implement bleeding-edge research (FP8 quantization, speculative decoding, advanced attention)
Ship optimizations to our open-source MAX platform that developers worldwide depend on
Qualification
Required
Currently pursuing a Bachelor's, Master's, or PhD in Computer Science, Engineering, or related field with graduation expected by Spring 2027 at the latest
Strong Python programming skills and ML/Deep Learning coursework or projects (PyTorch, TensorFlow, etc.)
Passion for performance optimization and systems programming with curiosity for solving complex problems
Strong verbal and written communication skills, ability to collaborate with mentors and peers
Preferred
Helpful experience includes: systems programming (C++, Rust), CUDA/ROCm, Python multiprocessing/asyncio, compilers, profiling tools, or parallel computing concepts
Benefits
Competitive Compensation
Stock options
Team Building Events
Company
Modular
Modular provides AI infrastructure for deployment, serving, and programming GPUs.
Founded in 2022
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
General Catalyst
2026-01-14
General Catalyst
2026-01-14
Greylock
2025-12-29
Company data provided by crunchbase