What you’ll be doing:
In this role you will work closely with deep learning compiler developers to create software infrastructure and applications used to develop, test, and release compilers. Responsibilities involve creating and maintaining advanced CI/CD pipelines to deploy and monitor machine learning software workloads. This includes functional and performance benchmarking solutions used to verify and bring the quality of deep learning compilers at NVIDIA to the next level. In addition, you will evaluate the performance of AI models and systems by implementing and utilizing advanced algorithms and solutions to carry out meticulous testing, experimentation, and benchmarking. You will help identify potential or observed weaknesses in the current process, offer ideas for actions that can improve code coverage, and participate in quality initiatives and drive continuous improvement.
What we need to see:
BS, MS, or PhD or equivalent experience in Computer Science, Computer/Electrical Engineering, Mathematics or related field
5+ years of overall professional experience with 2+ years of MLOps experience specific in deep learning software environment
Knowledge of the machine learning domain and professional experience working with deep learning frameworks like TensorRT, JAX, PyTorch, NeMo, TensorFlow, or others.
Able to architect, design, implement and debug complex solutions using Python programming language in machine learning domain
Familiarity with CI/CD pipeline development, cloud-based testing (AWS/GCP), data storage and database solutions, implementing visualization dashboards, and data analysis for performance and regression reporting
Learn and adapt swiftly to changes in technical and project requirements
Ways to stand out from the crowd:
Experience identifying and solving challenging technical problems by programmatic use of deep learning solutions such as large language models
Prior experience providing technical leadership in deep learning software development environments
Knowledge on how compilers work and how compilers are implemented with LLVM infrastructure
You will also be eligible for equity and .
משרות נוספות שיכולות לעניין אותך