Senior Software Engineer, FlashInfer and cuDNN Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 2 days ago

Senior Software Engineer, FlashInfer and cuDNN Infrastructure

NVIDIA is seeking excellent software engineers to enable the next wave of NVIDIA’s highest performing deep learning libraries. The role focuses on designing and developing scalable infrastructure that streamlines development and testing across NVIDIA’s diverse platforms. Responsibilities include building automation for integration and release processes, and advancing innovative tools in the open-source community.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Designing and developing software for testing and analysis of our codebases
Building scalable automation for build, test, integration, and release processes for publicly distributed deep learning libraries
Developing throughout the software stack, from the user experience down to the cluster and database layers
Configuring, maintaining, and building upon deployments of industry-standard tools (e.g. Kubernetes, Jenkins, Docker, CMake, Gitlab, Jira, etc)
Advancing innovative in those industry-standard tools and upstreaming contributions to the open source community

Qualification

PythonC/C++Continuous IntegrationGitDockerKubernetesJenkinsAutomationCollaborationProblem Solving

Required

BS or equivalent experience or higher degree in Computer Science or Computer Engineering with 5+ years of relevant experience
Strong programming skills in Python (or similar) and familiarity with C/C++ development
Experience setting up, maintaining, and automating continuous integration systems
Proficiency in SCM (e.g. Git, Perforce) and build systems (e.g. Make, CMake, Bazel)
A pragmatic approach to solving problems collaboratively with a passion for 'it just works' automation to enable team members

Preferred

Experience designing and developing automation in Jenkins, Gitlab CI/CD, or Github Actions and background with distributed systems and cluster/cloud computing (e.g. Slurm, containers, Kubernetes, etc)
Experience designing and developing unit and integration test frameworks with hands-on experience using code coverage and static code analysis tools
Success leading a team of engineers and/or experience as an active contributor to a software project involving many developers
Knowledge of GPU computing systems and experience with mobile/embedded platforms and multiple operating systems (Ubuntu, CentOS, Windows, L4T, or similar)
Track record of identifying useful new technologies and incorporating them into SW development flows

Benefits

Equity
Benefits

Company

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

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
Company data provided by crunchbase