200+ applicants

Company

NVIDIA · 1 day ago

Software Engineering Intern, JAX - Summer 2025

Santa Clara, CA

Internship

Onsite

Intern

$18/hr - $71/hr

Start in 2025 Summer

Maximize your interview chances

Artificial Intelligence (AI)GPU

Growth Opportunities

H1B Sponsor Likely

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

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

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

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

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10,001+ employees

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

2023 (735)

2022 (892)

2021 (696)

2020 (534)