Infrastructure Software Engineer, Deep Learning Libraries
Nvidia · Shanghai
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!