Senior Engineer, Math Libraries - Consumer and Embedded Platforms
$152k - $288k • US, CA, Santa Clara
Posted 14d ago
Job Location
US, CA, Santa Clara
Tech Stack
Remote Work Policy
On-site
Categories
Machine Learning Engineer
About the job
We are seeking a Senior Software Engineer to join our team focused on enabling CUDA Math Libraries on new and specialized platforms. This role is crucial for ensuring that NVIDIA's math libraries deliver high functionality and performance across a wide range of targets, supporting the AI, data analytics, and scientific simulation industries. You will be responsible for the integration, functional qualification, compliance, and performance readiness of these libraries on emerging platforms, collaborating closely with core library and devops engineering teams. This is an exciting opportunity to contribute to the forefront of application performance in the AI era.
Responsibilities
- Ensure platform readiness for CUDA Math Libraries on new and specialized platforms.
- Triage and isolate platform-specific issues, fixing bugs for functional correctness.
- Identify performance targets for new platforms, establish performance testing, and resolve regressions.
- Collaborate with CI/CD, build, and test infrastructure teams to create platform-specific qualification pipelines.
Requirements
- PhD or MSc degree in Computational Science and Engineering, Computer Science, Applied Mathematics, or related field (or equivalent experience).
- 5+ years of experience developing high-performance numerical software.
- Experience with object-oriented programming, large system software architecture, parallel computing, testing, maintenance, and performance optimization of HPC software using C++ and Python.
- Strong understanding of fundamental numerical methods and computations in science, engineering, and/or deep learning.
- Proficiency in software development practices, including CI/CD systems and project management tools like JIRA.
- Strong communication, collaboration, and documentation skills.
Benefits
- Equity
- Benefits