Infrastructure Software Engineer, Deep Learning Libraries

NVIDIA·Shanghai, Shanghai, China | Beijing, Beijing, China·posted 85d ago · last seen 55m ago

Track this application

Get Started Free

Match score against your CV

Get Started Free

Tailor your resume to this job

Get Started Free

About interviewing at NVIDIA

Recruiter screen, a technical screen mixing resume deep-dive with live coding, a hiring-manager conversation, then a panel of three to five 45–60 minute rounds: coding, systems design under hardware constraints, a domain deep-dive, and behavioral. Highly team-specific — you interview directly with the team — with C++ depth expected almost universally and decisions sometimes taking five-plus weeks after the panel.

Read the full NVIDIA interview process →

Description

We are now looking for an Infrastructure Software Engineer for Deep Learning Libraries!

NVIDIA's Deep Learning Libraries Group is seeking excellent software engineers to enable the next wave of NVIDIA’s highest performing deep learning libraries. The role focuses on NVIDIA's open-source products such as CUTLASS. The mission is to design and develop scalable, modular infrastructure that streamlines development, builds, and tests across NVIDIA’s diverse set of platforms, and address the needs from the open-source community, with the cutting-edge AI technology. Join our technically diverse team of software engineers and infrastructure experts to design the systems that enable NVIDIA to stay ahead of the competition as we deliver the world's fastest deep learning platforms.

What you'll be doing:

  • Designing and developing software for testing and analysis of our codebases

  • Building scalable automation for build, test, integration, and release processes for open-source products

  • Developing and deploying AI agents and similar technology to automate the end-to-end software development cycle

  • Configuring, maintaining, and building upon deployments of industry-standard tools (e.g. Kubernetes, Jenkins, Docker, CMake, Gitlab, Jira, etc.)

  • Advancing the state of the art in those industry-standard tools

What we need to see:

  • A Masters Degree in Computer Science or Computer Engineering or equivalent experience.

  • 3+ 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 (e.g. Jenkins, GitHub Actions, GitLab pipelines, Azure DevOps)

  • Extensive experience in AI agents technology

  • Fluency in SCM (e.g. Git, Perforce) and build systems (e.g. Make, CMake, Bazel)

Ways to stand out from the crowd:

  • Experience designing and developing automation in Jenkins with Groovy (or similar)

  • Background with distributed systems and cluster/cloud computing, especially with Kubernetes

  • Experience designing and developing unit and integration test frameworks

  • Close follow the latest trend in AI industry

  • Track record of identifying useful new technologies and incorporating them into SW development flows

This is an opportunity to have a wide impact at NVIDIA by improving development velocity across our many AI/DL/Compute Software projects. Are you creative, driven, and autonomous? Do you love a challenge? If so, we want to hear from you!

More engineering roles at NVIDIA