NVIDIA is looking to optimize Machine Learning training, inference, and NVIDIA AI Services by squeezing every last clock cycle out of their systems, and is seeking senior engineers to help achieve peak performance across the hardware/software stack.
Requirements
- Excellent Python programming skills, extensive knowledge of at least one ML Framework
- Algorithms and ML fundamentals
- Experience with Git or other source-control systems.
- Strong problem solving and analytical skills
- Extensive knowledge of ML frameworks such as PyT, JAX, TF2, DGL and others
Responsibilities
- Implement machine learning models from multiple data domains (CV, NLP/LLMs, ASR, TTS, RecSys and others) in multiple ML frameworks (PyT, JAX, TF2, DGL and others)
- Implement and test new SW features (Graph Compilation, reduced precision training) that use the most recent HW functionalities.
- Analyze, profile, and optimize machine learning workloads on pioneering hardware and software platforms.
- Collaborate with researchers and engineers across NVIDIA, providing guidance on improving the design, usability and performance of models.
- Lead best-practices for building, testing, and releasing ML software
Other
- BSc, MS or PhD degree in Computer Science, Computer Architecture, Mathematics, Physics or related technical field or equivalent experience
- 8+ years of experience in ML model implementation and SW Development
- Published papers, patents or articles
- Ability to work in a diverse, supportive environment
- Must be eligible to work in the US
