Software Engineering Intern, JAX - Summer 2025 jobs in United States
info-icon
This job has closed.
company-logo

NVIDIA · 1 year ago

Software Engineering Intern, JAX - Summer 2025

positionSanta Clara, CA
timeInternship
remoteOnsite
seniorityIntern
money$18/hr - $71/hr

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. They are seeking a Software Engineering Intern to contribute to the development of performance optimizations for deep learning frameworks using JAX, focusing on enhancing the efficiency of JAX-based systems for real-world applications.

Artificial Intelligence (AI)Consumer ElectronicsGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Make meaningful contributions to JAX by designing and implementing core components that drive phenomenal performance on the NVIDIA AI platform
Increase the efficiency of teams developing JAX-based systems by develop tools that streamline the application of deep learning research and simulations to real-world products

Qualification

JAXPythonC++Machine Learning FrameworksDistributed SystemsCPU ArchitecturesGPU ArchitecturesNumeric LibrariesModular Software Design

Required

Pursuing a MS or PhD in Computer Science, Computer Engineering or related field
Understanding of JAX, Autograd, tracing, code generation and DSL compilers
Working proficiency in Python, familiarity with C++
Knowledge of machine learning frameworks (e.g., PyTorch, TensorFlow, scikit-learn)

Preferred

Familiar with distributed systems, services, or Deep Learning at large scale
Ability to work optimally with multi-functional teams
Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design

Benefits

Intern benefits

Company

NVIDIA

twittertwittertwitter
Glassdoor4.6
company-logo
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
dateFounded in 1993
location
Santa Clara, California, USA
people-size10001+ employees
web-link
https://www.nvidia.com

H1B Sponsorship

NVIDIA 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 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin

Recent News

MarketScreener
Nvidia CEO Huang touts automotive AI at CES talk as competition mounts
2026-01-06
MarketScreener
Nvidia: Rubin-based products will be available from second half of 2026
2026-01-06
GlobeNewswire
NVIDIA Announces Alpamayo Family of Open-Source AI Models and Tools to Accelerate Safe, Reasoning-Based Autonomous Vehicle Development
2026-01-06
Company data provided by crunchbase