NVIDIA is looking for a Manager, ML Engineering to lead and grow the team responsible for building and optimizing accelerated machine learning tools and analytics pipelines on GPUs. In this role, you will provide technical direction, mentorship, and collaborate with partners to advance the state of ML on GPUs.
What You'll Do
- Lead, mentor, and grow your engineering team, overseeing project planning, execution, and the quality of our ML libraries.
- Work closely with Product, Project Management, and external partners to develop ML applications that benefit the broader ML community and enterprise partners.
- Define success and evaluation criteria, including establishing reproducible workflows and datasets for library development and benchmarking.
- Provide technical leadership and mentorship to engineers contributing across different teams and projects.
What We're Looking For
- A Master's or PhD degree in Computer Engineering/Science, or equivalent experience in a related field.
- 8+ years of overall experience in Computer Science, Artificial Intelligence, Applied Math, or a related field.
- 3+ years of experience recruiting, training, and leading software engineering teams.
- Strong written communication, collaboration, and presentation skills.
- Experience across the full ML lifecycle: from research and training to tuning and deploying models.
- Strong analytical problem-solving skills and fundamentals in algorithms and mathematics.
- Experience with distributed system design and development.
- Experience designing, developing, benchmarking, and tuning complex multi-language software stacks (C++/CUDA and Python).
Nice to Have
- Experience working closely with partners and customers to build successful ML-based tooling.
- Background with Vector Search and Databases.
- Experience contributing to and/or maintaining open-source software projects.
- Experience developing distributed algorithms and running on distributed systems such as HPC or Cloud.
- Experience with PyData tools: NumPy, Pandas, Scikit-Learn, Dask, Ray, Xarray, Zarr.
Technical Stack
- C++, CUDA, Python
- PyData: NumPy, Pandas, Scikit-Learn, Dask, Ray, Xarray, Zarr
Benefits & Compensation
- Compensation: $224,000 - $356,500 USD for Level 3, and $272,000 - $425,500 USD for Level 4.
- Eligible for equity.
- Comprehensive benefits.
NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
