AI Kernel Engineering Intern jobs in United States
cer-icon
Apply on Employer Site
company-logo

Modular · 9 hours ago

AI Kernel Engineering Intern

Modular is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack. As an AI Kernel Engineering Intern, you will work on designing, optimizing, and benchmarking GPU kernels for AI workloads, gaining hands-on experience with AI frameworks and performance optimization.

AI InfrastructureArtificial Intelligence (AI)Generative AIMachine LearningSoftware
check
H1B Sponsor Likelynote

Responsibilities

Help design, optimize, and benchmark low-level kernels that power modern AI workloads
Focus on GPUs and emerging accelerators
Projects may include optimizing matrix operations, attention primitives, or custom operators; analyzing memory layouts and data movement; and collaborating with others to translate models into efficient, production-ready implementations

Qualification

Parallel programmingPerformance optimizationMachine learning fundamentalsPythonC/C++CUDACuriousDetail-orientedExcited in fast-paced environment

Required

Currently pursuing a Bachelor's, Master's, or PhD in Computer Science, Math, Electrical Engineering, or a related field
Strong foundation in parallel programming, performance optimization, memory subsystem, or computer architecture
Machine learning fundamentals, AI inference models, and modern AI workloads
Proficiency in Python, C/C++, or any object-oriented languages
Experience with CUDA, or other accelerator programming models

Preferred

A publication record is a nice-to-have
Curious, detail-oriented, and excited in a fast-paced startup environment

Benefits

Competitive Compensation
Team Building Events

Company

Modular

twittertwittertwitter
company-logo
Modular provides AI infrastructure for deployment, serving, and programming GPUs.

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 Stage
Total Funding
$380M
Key Investors
US Innovative Technology FundGeneral CatalystGoogle Ventures
2025-09-24Series C· $250M
2023-08-24Series B· $100M
2022-06-30Seed· $30M

Leadership Team

leader-logo
Chris Lattner
CEO + Co-Founder
linkedin
leader-logo
Tim Davis
Co-Founder & President
linkedin

Recent News

Company data provided by crunchbase