NVIDIA · 1 week ago
Senior Software Engineer, PyTorch - Deep Learning
United States
Full-time
Remote
Mid, Senior Level
$152K/yr - $218K/yr
3+ years expNVIDIA is a leading technology company specializing in AI applications and deep learning. They are seeking a Senior Deep Learning Software Engineer, PyTorch to design and build tools for AI engineers, contributing to the PyTorch framework and collaborating with various teams to enhance GPU performance.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
Hiring Manager
Bushra S.
Responsibilities
Design and build PyTorch components that run efficiently on supercomputers with 1000s-100ks of GPUs
Collaborate with NVIDIA’s hardware and software teams to improve the overall GPU performance in PyTorch
Design, build and support production AI solutions used by enterprise customers and partners
Work with internal applied researchers to improve their AI tools
Qualification
Required
BS in Computer Science or Engineering (or equivalent experience)
3+ years professional experience in deep learning
Proficient with C++ programming
Strong understanding of systems software and interfaces
Demonstrated experience with Thread and Distributed Parallel Programming
Demonstrated background developing large software projects
Strong verbal and written communication skills
Preferred
Contributions and participation in the open source community
Familiarity with deep learning compilers
Familiarity with deep learning modeling trends
Background with CUDA Programming as well as Python
Demonstrated experience working with multi-disciplinary teams
Benefits
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
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 CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
The Motley Fool
2026-02-04
2026-02-04
2026-02-04
Company data provided by crunchbase