Want to professionalize your AI skills, pivot to an AI role and increase your salary?
Master AI Engineering with the most practical and comprehensive LLM Development certifications at Towards AI Academy.

NVIDIA

Software Engineering Intern, JAX - Fall 2026

NVIDIA

Published 25 Apr 2026
Santa Clara, CA, USA
Full Time

Share this job

Role Highlights

Languages used

Python
C++

Key skills

Machine Learning
Deep Learning
ML Research
Computer Science
Written Communication
Distributed Systems
Software Design
AI
PhD
Compilers
CPU
Infrastructure
GNNs

Tools, Libraries and Frameworks

GPUs
JAX
XLA
PyTorch
Tensorflow
Scikit-learn

Description

The role involves developing and deploying machine learning models. The individual will be responsible for building data pipelines and infrastructure to support these models. They will also contribute to the research and implementation of new AI techniques. The position requires collaboration with cross-functional teams to integrate AI solutions into existing products.

Required Qualifications and Skills

The role requires proficiency in Python and SQL for data manipulation and model development. Experience with cloud platforms such as AWS is necessary. Familiarity with containerization technologies like Docker and orchestration tools like Kubernetes is also required. The position necessitates expertise in machine learning frameworks such as TensorFlow, PyTorch, and Scikit-learn. A strong understanding of MLOps principles and data engineering practices is essential.

Disclaimer

Disclaimer: Job and company description information and some of the data fields may have been generated via GPT-4 summarisation and could contain inaccuracies. The full external job listing link should always be relied on for authoritative information.

About the company

NVIDIA

Size

27685

Founded

HQ

Santa Clara, US

Public/Private

Public Company

Description

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Share

Share this job

Related jobs

Integrations
Product Development
Test Engineer
Computer Graphics
Santa Clara, CA, USA
Full Time
Computer Science
Integrations
Product Management
Electrical Engineering
Santa Clara, CA, USA
Full Time
Computer Science
Integrations
Electrical Engineering
Program Management
Santa Clara, CA, USA
Full Time