Description
We're seeking software engineers for our Deep Learning Frameworks Sustaining Engineering team at NVIDIA. You'll work on software used globally, collaborating with deep learning framework developers to integrate with open-source codebases.
In this role, you'll:
- Back-port changes from the mainline branch
- Track open-source dependency changes
- Ensure the latest stable dependencies are used for enterprise products
- Contribute to timely Long Term Support releases for TensorFlow, PyTorch, and TensorRT
- Fix customer-reported bugs and integrate bug fixes from mainline
- Work with teams to patch open-source dependencies for security vulnerabilities
Requirements:
- Bachelor's in Computer Science, Deep Learning, AI, Applied Math, or related field
- 5+ years of software development experience
- Excellent C/C++ programming and software design skills
- Experience with tools like Make, Docker, Bazel, Debian, pip, npm, Gitlab, and CI/Jenkins
- Prior experience with machine learning algorithms and frameworks (TensorFlow, PyTorch, MXNet)
- Ability to work independently and communicate effectively
Preferred qualifications:
- GPU programming experience (CUDA or OpenCL)
- Experience with large open-source projects and Gitlab CI pipelines
We offer a competitive salary, equity, and benefits. Expected base salary ranges are $152,000 - $241,500 for Level 3 and $184,000 - $287,500 for Level 4.