1

Pytorch Internship Jobs in California (NOW HIRING)

Agentic AI Engineer

Santa Clara, CA · On-site

$135K - $162K/yr

... or PyTorch). • Academic or internship experience in machine learning, AI, or data-driven projects. • Solid understanding of algorithms, data structures, and machine learning concepts. • ...

Senior Machine Learning Engineer

San Francisco, CA · On-site

$144K - $190K/yr

Required : • 4+ years of non-internship professional MLE experience. • Deep expertise in ... Fluency in PyTorch or JAX for training large-scale models. • Proficiency in Python and ...

next page

Showing results 1-20

Pytorch Internship information

What types of projects and collaborative experiences can I expect during a PyTorch Internship?

During a PyTorch Internship, you can expect to work on hands-on machine learning and deep learning projects that involve developing, testing, and optimizing models using the PyTorch framework. Interns often collaborate closely with research scientists, software engineers, and product teams to contribute to real-world applications and open-source initiatives. You may participate in code reviews, brainstorming sessions, and weekly progress meetings, gaining exposure to both independent tasks and team-based problem-solving. This environment fosters both technical growth and communication skills, preparing you for advanced roles in AI and machine learning.

What are the key skills and qualifications needed to thrive as a PyTorch Intern, and why are they important?

To thrive as a PyTorch Intern, you need a solid background in Python programming, machine learning fundamentals, and familiarity with deep learning concepts, typically evidenced by coursework or project experience. Proficiency in PyTorch, version control systems like Git, and tools such as Jupyter Notebooks is highly valued. Strong problem-solving skills, attention to detail, and effective communication help interns contribute meaningfully to team projects and learn quickly. These skills and qualities are crucial for efficiently developing, testing, and deploying machine learning models in a collaborative environment.

What is the difference between Pytorch Internship vs Machine Learning Intern?

AspectPytorch InternshipMachine Learning Intern
Required SkillsProficiency in Pytorch, Python, deep learning conceptsPython, machine learning algorithms, data analysis
Work EnvironmentResearch labs, tech companies, AI startupsTech firms, research institutions, data-driven companies
Industry UsageDeep learning projects, neural network developmentBroader ML applications, data modeling

Both roles involve working with machine learning, but a Pytorch Internship specifically focuses on deep learning frameworks like Pytorch, while a Machine Learning Intern may work across various ML techniques. The Pytorch Internship is ideal for those specializing in neural networks and deep learning, whereas the Machine Learning Intern role covers a wider range of ML applications.

What is a PyTorch internship?

A PyTorch internship is a temporary position, often for students or recent graduates, where individuals gain hands-on experience working with the PyTorch deep learning framework. Interns typically assist with machine learning projects, develop and test models, and contribute to research or product development involving artificial intelligence. These internships provide valuable exposure to real-world applications of AI, opportunities to collaborate with experienced engineers and researchers, and a chance to enhance programming and problem-solving skills. Many internships also offer mentorship and may lead to full-time roles in the field.
What are the most commonly searched types of Pytorch jobs in California? The most popular types of Pytorch jobs in California are:
What cities in California are hiring for Pytorch Internship jobs? Cities in California with the most Pytorch Internship job openings:
PhD Research Intern, Generative AI - 2026

PhD Research Intern, Generative AI - 2026

Nvidia

Santa Clara, CA

Full-time

Posted 7 days ago


Job description

AtNVIDIA, we are building Cosmos world foundation models and generative AI systems for Physical AI across robotics, autonomous driving, smart spaces, and embodied agents.

TheNVIDIA Cosmos Platformenables multimodal world understanding, simulation, synthetic data generation, and embodied reasoning. We are looking for outstanding PhD interns to help advance the frontier of Physical AI and world models.

What you'll be doing:

  • Conduct research in generative AI, multimodal foundation models, world models, and embodied AI.

  • Develop algorithms for video understanding/generation, action-conditioned simulation, multimodal reasoning, and policy learning.

  • Train and evaluate large-scale models using video, image, language, and robotics or autonomous driving data.

  • Collaborate with researchers and engineers across AI, robotics, simulation, and graphics teams.

  • Publish research at top conferences and transfer innovations into NVIDIA products.

What we need to see:

  • Currently pursuing a PhD in CS, EE, Robotics, or related fields.

  • Strong background in generative AI, computer vision, multimodallearning, robotics, or reinforcement learning.

  • Prior publication record and research experience.

  • Strong Python and PyTorch skills.

Ways to stand out from the crowd:

  • Experience with large-scale foundation model training.

  • Research in video models, VLMs, world models, robotics, or autonomous driving.

  • Experience with distributed training, simulation, or embodied AI.

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you!

Our internship hourly rates are a standard pay based on the position, your location, year in school, degree, and experience. The hourly rate for our interns is 30 USD - 94 USD.

You will also be eligible for Internbenefits.

Applications for this job will be accepted at least until May 31, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

Nvidia logo

About Nvidia

Sourced by ZipRecruiter

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology--and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.

Industry

Computer and electronic product manufacturing

Company size

10,000+ Employees

Headquarters location

Santa Clara, CA, US

Year founded

1993