Machine Learning Software Engineer - New College Grad 2025
NVIDIA
On-site 🏢 New 🔥🔥
Data Science & AI/ML
California, United States 🇺🇸
At NVIDIA, we’re building the foundation for the next generation of Physical AI. Our Cosmos team is passionate about developing robust, production-grade open-source Machine Learning software to support machine learning and generative-AI research at scale. We’re looking for expert-level Python engineers who are passionate about building production-ready systems and want to make a lasting impact through open-source contributions.
If you care deeply about software craftsmanship, maintainability, and performance—and have hands-on experience building ML systems—this role is for you.
What you’ll be doing:
Develop and maintain high-quality, modular, and well-tested Python code for large-scale ML infrastructure. See https://github.com/nvidia-cosmos
Design and optimize post-training, inference, and data processing pipelines used by ground breaking ML models.
Collaborate with research and product teams to bring ML systems from prototype to production.
Contribute to open-source projects and build internal tools that enable scalable AI experimentation.
Improve performance, reliability, and observability of large distributed systems.
Support teammates through design reviews, code reviews, and collaborative development.
What we need to see:
Pursuing BS, MS or PhD degree in Computer Science, Engineering, or a related field, or equivalent experience
Strong proficiency in Python and a track record of delivering production-quality software.
Experience with PyTorch (or similar frameworks such as JAX or TensorFlow), especially in real-world applications.
Deep understanding of ML system design, training loops, data loaders, evaluation, and model serving.
Familiarity with containerization, CI/CD, and maintaining in production environments
Comfortable working with large codebases, building reusable libraries, and writing documentation and tests.
Ways to stand out in the crowd:
Contributions to open-source ML or Python infrastructure projects.
Background in scaling ML training and inference systems across GPUs as well as experience building libraries that wrap or extend PyTorch functionality.
Prior exposure to multimodal models or simulation environments (vision, language, audio).
Familiarity with NVIDIA’s GPU compute stack or high-performance computing clusters.
Experience with distributed computing tools like DDP, FSDP, ZeRO, or Ray.
Are you dedicated, upbeat and dynamic with excellent analytical ability? Are you an engineer passionate and highly motivated about solving complex problems? If so, you may be a perfect fit for NVIDIA!
You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.
Similar jobs
Machine Learning Intern - 2025
On-site 🏢 30 January
Data Science & AI/ML
STP, Hong Kong 🇭🇰
Data Science Intern
On-site 🏢 09 March
Data Science & AI/ML
Singapore 🇸🇬
2025 Applied Science Internship - Recommender Systems/ Information Retrieval (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 02 October
Data Science & AI/ML
Seattle, United States 🇺🇸
Business Analytics Intern (Sourcing Team，深圳，2025实习)
On-site 🏢 21 October
Data Science & AI/ML
Shenzhen, China 🇨🇳
Machine Learning / Artificial Intelligence Intern/Co-Op (PhD | Fall 2025 | Hybrid)
On-site 🏢 12 April
Data Science & AI/ML
California, United States 🇺🇸
AI Algorithms Software Engineer (RDSS Intern)
On-site 🏢 23 October
Data Science & AI/ML
Hsinchu, Taiwan 🇹🇼 Taipei, Taiwan 🇹🇼
ML Compiler Software Engineer PEY Co-op (12-16 months), Annapurna ML
On-site 🏢 10 September
Data Science & AI/ML
Toronto, Canada 🇨🇦
Machine Learning / Artificial Intelligence Intern/Co-Op (PhD | Fall 2025 | Hybrid)
On-site 🏢 07 April
Data Science & AI/ML
Texas, United States 🇺🇸
Deep Learning Intern - Fall 2025
On-site 🏢 22 April
Data Science & AI/ML
Shanghai, China 🇨🇳
AI Machine Learning Engineer II (Full Time) United States
Remote 🌎🌍🌏 29 April
Data Science & AI/ML
California, United States 🇺🇸
Data Scientist intern
On-site 🏢 09 April
Data Science & AI/ML
Singapore 🇸🇬
2025 Applied Scientist Internship, Amazon University Talent Acquisition
On-site 🏢 01 November
Data Science & AI/ML
Amsterdam, Netherlands 🇳🇱
2025 Software Development Engineer Intern - Machine Learning (m/w/d)
On-site 🏢 01 November
Data Science & AI/ML
Berlin, Germany 🇩🇪
Research Scientist L4/L5, Algorithms Engineering
Remote 🌎🌍🌏 11 January
Data Science & AI/ML
United States 🇺🇸
Machine Learning / Artificial Intelligence Intern/Co-Op (Graduate | Fall 2025 | Hybrid)
On-site 🏢 07 April
Data Science & AI/ML
Texas, United States 🇺🇸
Fall 2025 Applied Science Internship - Recommender Systems/ Information Retrieval (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 New 🔥🔥
Data Science & AI/ML
Seattle, United States 🇺🇸
Estágio em Engenharia de Dados
On-site 🏢 07 May
Data Science & AI/ML
Campinas, Brazil 🇧🇷
Data Engineering Intern
On-site 🏢 09 March
Data Science & AI/ML
Singapore 🇸🇬
大模型LLM算法实习生（推理优化-算法）(Jul - Dec 2025)
On-site 🏢 16 April
Data Science & AI/ML
Beijing, China 🇨🇳
Business Intelligence Engineer, Amazon Logistics
On-site 🏢 07 March
Data Science & AI/ML
Tokyo, Japan 🇯🇵
Data Scientist Intern
On-site 🏢 28 April
Data Science & AI/ML
Singapore 🇸🇬
Machine Learning Intern, Fall 2025
On-site 🏢 23 April
Data Science & AI/ML
California, United States 🇺🇸
Machine Learning Engineer (L5 - Senior) , Ads Inventory Management & Forecasting
Remote 🌎🌍🌏 08 April
Data Science & AI/ML
United States 🇺🇸
NLP Research Intern
On-site 🏢 07 January
Data Science & AI/ML
Singapore 🇸🇬
Research Intern - Database Internals (PostgreSQL)
Hybrid 🏡🏢 11 March
Data Science & AI/ML
Washington, United States 🇺🇸
AI Computing Architect Intern - 2025
On-site 🏢 New 🔥🔥
Data Science & AI/ML
Shanghai, China 🇨🇳
Machine Learning Software Engineer L4/L5
Remote 🌎🌍🌏 09 January
Data Science & AI/ML
United States 🇺🇸
Data Science Intern
On-site 🏢 09 March
Data Science & AI/ML
Singapore 🇸🇬
Intern as AI Engineer - part time for student
On-site 🏢 17 March
Data Science & AI/ML
Budapest, Hungary 🇭🇺
WeChat Tech Intern (Data Scientist - Safety Strategies)
On-site 🏢 22 August
Data Science & AI/ML
Singapore 🇸🇬
Machine Learning Software Engineer (L5) - Promo Media & Marketing
Remote 🌎🌍🌏 14 May
Data Science & AI/ML
United States 🇺🇸
- Machine Learning Development Intern
On-site 🏢 20 May
Data Science & AI/ML
Tokyo, Japan 🇯🇵
Machine Learning Engineer
Remote 🌎🌍🌏 08 January
Data Science & AI/ML
United States 🇺🇸
Machine Learning / Artificial Intelligence Intern/Co-Op (Undergraduate | Fall 2025 | Hybrid)
On-site 🏢 12 April
Data Science & AI/ML
California, United States 🇺🇸
Machine Learning Scientist (L5) - Member Lifecycle and Monetization DSE
Remote 🌎🌍🌏 03 February
Data Science & AI/ML
United States 🇺🇸
大模型LLM算法实习生 - 模型量化/低比特训练推理 (Jul - Dec 2025)
On-site 🏢 17 April
Data Science & AI/ML
Beijing, China 🇨🇳
NLP/LLM Research Intern
On-site 🏢 15 November
Data Science & AI/ML
London, United Kingdom 🇬🇧
2025 Business Intelligence Engineer Internship
On-site 🏢 09 October
Data Science & AI/ML
Istanbul, Türkiye 🇹🇷
Cambridge Research Intern – AI Infrastructure & Machine Learning
Hybrid 🏡🏢 New 🔥🔥
Data Science & AI/ML
Cambridgeshire, United Kingdom 🇬🇧
Applied Machine Learning Intern
On-site 🏢 20 May
Data Science & AI/ML
Palo Alto, United States 🇺🇸
图像视频生成算法实习生 ( Jul - Dec 2025)
On-site 🏢 16 April
Data Science & AI/ML
Beijing, China 🇨🇳
AI Solution (Intern) - Singapore
On-site 🏢 13 May
Data Science & AI/ML
Singapore 🇸🇬
Data Science Intern
On-site 🏢 10 March
Data Science & AI/ML
Amsterdam, Netherlands 🇳🇱
Data Science Intern
On-site 🏢 09 March
Data Science & AI/ML
Amsterdam, Netherlands 🇳🇱
2025 Applied Science Internship - Information & Knowledge Management (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 02 October
Data Science & AI/ML
Seattle, United States 🇺🇸
大模型LLM算法实习生- 模型加速 (Jul - Dec 2025)
On-site 🏢 11 April
Data Science & AI/ML
Beijing, China 🇨🇳
2025, Data Analyst Intern- Community Operations, Japan
Hybrid 🏡🏢 07 May
Data Science & AI/ML
Tokyo, Japan 🇯🇵
Praktikant*in (m/w/d) Analytics & Insights
Hybrid 🏡🏢 12 March
Data Science & AI/ML
Eschborn, Germany Home Office
NLP Research Intern
On-site 🏢 20 May
Data Science & AI/ML
Tokyo, Japan 🇯🇵
Fall 2025 Applied Science Internship - Information & Knowledge Management (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 New 🔥🔥
Data Science & AI/ML
Seattle, United States 🇺🇸
Deep Learning Architect - New College Grad 2025
On-site 🏢 New 🔥🔥
Data Science & AI/ML
North Carolina, United States 🇺🇸 California, United States 🇺🇸
Machine Learning Researcher Intern - Prague
On-site 🏢 17 April
Data Science & AI/ML
Stredocesky, Czech Republic 🇨🇿
Business Intel Engineer Intern, Prime International
On-site 🏢 28 April
Data Science & AI/ML
Singapore 🇸🇬
Machine Learning / Artificial Intelligence Intern/Co-Op (Graduate | Fall 2025 | Hybrid)
On-site 🏢 12 April
Data Science & AI/ML
California, United States 🇺🇸
AI/ML 测试开发实习生 (Jul - Dec 2025)
On-site 🏢 17 April
Data Science & AI/ML
Shanghai, China 🇨🇳
Business Analyst (Intern) Colombia
On-site 🏢 17 April
Data Science & AI/ML
Bogota, Colombia 🇨🇴
Business Analytics Intern (Program Team，深圳，2025实习)
On-site 🏢 27 December
Data Science & AI/ML
Shenzhen, China 🇨🇳
Data Scientist Intern
On-site 🏢 25 April
Data Science & AI/ML
Singapore 🇸🇬
Data Science Intern (ML/AI)
On-site 🏢 25 March
Data Science & AI/ML
Amsterdam, Netherlands 🇳🇱
Alternant(e) Data IA Ingénieur Google Cloud, (H/F)
Hybrid 🏡🏢 01 April
Data Science & AI/ML
Saint-Ouen, France 🇫🇷
大模型LLM算法实习生 - 推理模型 (Jul - Dec 2025)
On-site 🏢 17 April
Data Science & AI/ML
Beijing, China 🇨🇳
Machine Learning Engineer (L5) - Content and Studio
Remote 🌎🌍🌏 13 April
Data Science & AI/ML
United States 🇺🇸
大模型LLM算法实习生 -推理优化 -工程 (Jul - Dec 2025)
On-site 🏢 16 April
Data Science & AI/ML
Beijing, China 🇨🇳
Data Engineer Intern
On-site 🏢 09 March
Data Science & AI/ML
Singapore 🇸🇬
Agentic AI Applied Scientist Intern, Shanghai AI Lab
On-site 🏢 New 🔥🔥
Data Science & AI/ML
Shanghai, China 🇨🇳
Business Intelligence Engineer Intern , Amazon University Talent Acquisition
On-site 🏢 15 November
Data Science & AI/ML
Sao Paulo, Brazil 🇧🇷
Machine Learning Engineer (L5) - Studio Media Algorithms
Remote 🌎🌍🌏 06 May
Data Science & AI/ML
United States 🇺🇸
AI Solution (Intern) - Indonesia
On-site 🏢 13 May
Data Science & AI/ML
Jakarta, Indonesia 🇮🇩
AI Digital Human Development Intern - 2025
Remote 🌎🌍🌏 02 February
Data Science & AI/ML
Remote, Hong Kong 🇭🇰
Stage Data Analyst Junior (H/F) – 6 mois
Hybrid 🏡🏢 20 May
Data Science & AI/ML
Saint-Ouen, France 🇫🇷
算子开发实习生 AI/ML Kernel Optimization Intern (Jul - Dec 2025)
On-site 🏢 03 April
Data Science & AI/ML
Shanghai, China 🇨🇳
Machine Learning Platform Development Intern
On-site 🏢 02 March
Data Science & AI/ML
Singapore 🇸🇬
2025 Business Intelligence Engineer Internship
On-site 🏢 02 October
Data Science & AI/ML
Luxembourg 🇱🇺
数据科学实习生 (Jul-Dec 2025)
On-site 🏢 28 April
Data Science & AI/ML
SUZHOU, China 🇨🇳
2025 Applied Science Internship - Natural Language Processing and Speech Technologies - United States, PhD Student Science Recruiting
On-site 🏢 02 October
Data Science & AI/ML
Seattle, United States 🇺🇸
WeChat - Data Science Intern
On-site 🏢 21 May
Data Science & AI/ML
Singapore 🇸🇬
2025 Applied Science Internship - Reinforcement Learning & Optimization (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 02 October
Data Science & AI/ML
Seattle, United States 🇺🇸
Data Analyst Intern/Co-Op (Undergrad | Fall, 2025 | Onsite/Hybrid | 4 months)
On-site 🏢 09 May
Data Science & AI/ML
Ontario, Canada 🇨🇦
Machine Learning Chip Architect, Annapurna Labs
On-site 🏢 17 March
Data Science & AI/ML
Cupertino, United States 🇺🇸
Share this job, spread the word!
Similar jobs
Machine Learning Intern - 2025
On-site 🏢 30 January
Data Science & AI/ML
STP, Hong Kong 🇭🇰
Data Science Intern
On-site 🏢 09 March
Data Science & AI/ML
Singapore 🇸🇬
2025 Applied Science Internship - Recommender Systems/ Information Retrieval (Machine Learning) - United States, PhD Student Science Recruiting
On-site 🏢 02 October
Data Science & AI/ML
Seattle, United States 🇺🇸