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

Modular · 3 hours ago

AI Models Engineering Intern

Modular is on a mission to revolutionize AI infrastructure by rebuilding the AI software stack. The AI Models Engineering Intern will implement cutting-edge AI models using Python, translating generative AI models into efficient, production-ready implementations, and collaborating with the optimization team for performance enhancements.

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

Responsibilities

Implement cutting-edge AI models and architectures using Python on top of our MAX framework
Work on accelerating popular models that will be showcased on builds.modular.com
Translate state-of-the-art generative AI models into efficient, production-ready implementations
Work together with our optimization team to help ensuring optimal performance through hardware acceleration and advanced optimizations

Qualification

PythonGenerative AI principlesDebuggingModel implementationPyTorchHugging FaceLLMs

Required

2025 graduate or currently pursuing a Bachelor's or Master's degree in Computer/Information Science, Mathematics, or an adjacent field; exceptional self-taught candidates with equivalent experience will also be considered
Strong Python programming expertise
Advanced debugging capabilities
Understanding of generative AI principles and architectures
Ability to translate AI research, or an existing implementation in another framework, into code

Preferred

Experience with PyTorch
Experience with LLMs and their underlying transformer architecture
Familiarity with Hugging Face ecosystem
Background in model implementation/acceleration

Benefits

Competitive Compensation.
Stock options.
Team Building Events.
Regular team onsites and local meetups in Los Altos, CA.

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