Senior Math Libraries Engineers - Python APIs

2 Months ago • 3 Years + • Research & Development • $148,000 PA - $287,500 PA

Job Summary

Job Description

NVIDIA seeks a self-motivated Senior Math Libraries Engineer to design and develop high-performance Python APIs for their math libraries. The role involves crafting and developing APIs (e.g., linear algebra), integrating them into libraries and frameworks, conducting performance analysis and optimization on CPU/GPU architectures, and collaborating with product management and partners to define feature and performance requirements. Responsibilities include prototyping, performance tuning, working in an agile environment, and improving user experience. The ideal candidate possesses strong Python and C++ skills, experience with numerical methods, and a proven track record in high-performance computing.
Must have:
  • PhD or MSc in relevant field or equivalent experience
  • Excellent Python & C++ programming skills
  • Experience developing Python APIs
  • Experience with NumPy, CuPy, SciPy, JAX, PyTorch, TensorFlow, or Numba
  • 3+ years experience in high-performance application development
Good to have:
  • GPU programming experience (CUDA or OpenCL)
  • Knowledge of CPU/GPU architecture
Perks:
  • Equity
  • Benefits

Job Details

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of high-performant Python APIs for our math libraries. In the last decade, Python has become the de-facto leading programming language for engineers in AI and data science, and more recently in HPC and scientific computing. NVIDIA has been at the forefront of providing GPU accelerated Deep Learning frameworks. These frameworks provide an efficient high-level programming interface allowing their users to focus on their application while providing highly optimized implementations. As the capabilities of our math libraries grow and extend to new domains, we would like to make them accessible to all developers in the Python ecosystem. In this role, you would be joining our dynamic team to help building and improving APIs for NVIDIA Math Libraries supporting the whole ecosystem of Python-based libraries and frameworks.

What you will be doing:

  • crafting and developing Python APIs for math libraries (e.g., linear algebra),

  • prototyping integrations of developed APIs into targeted libraries and frameworks,

  • conducting performance analysis and benchmarking, performance tuning and optimization, of numerical software on various CPU and GPU architectures,

  • working closely with product management and other internal and external partners to understand feature and performance requirements and contribute to the technical roadmaps of NVIDIA math libraries,

  • finding opportunities to improve target user experiences.

What we need to see:

  • PhD or MSc degree in Computer Science, Applied Math, or related science or engineering field is preferred or equivalent experience.

  • Excellent Python and C++ programming and software design skills, including functional and performance tests design.

  • Experience developing Python APIs in pythonic style

  • Experience using one or more of Numpy, CuPy, SciPy, JAX, PyTorch, TensorFlow, Numba.

  • 3+ years experience developing, debugging, and optimizing high-performance applications on parallel computing platforms.

  • Strong understanding of fundamental numerical methods and computations in science, engineering, and/or deep learning.

  • Proven experience in working on software development projects in an agile environment.

  • Strong collaboration, communication, and documentation habits.

Ways to stand out from the crowd:

  • Parallel programming experience with GPUs (CUDA or OpenCL), multi-threading, or MPI.

  • Good knowledge of CPU and/or GPU hardware architecture.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and have initiative, we want to hear from you!

The base salary range is 148,000 USD - 287,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

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.

Similar Jobs

Virtuos - Senior Games Tool Engineer (Machine Learning Specialist)

Virtuos

Shanghai, Shanghai, China (On-Site)
1 Month ago
Pluto7 - Data Scientist

Pluto7

Bengaluru, Karnataka, India (On-Site)
8 Months ago
Welevel - Senior AI Engineer (LLM & Training Focus)

Welevel

Munich, Bavaria, Germany (On-Site)
1 Month ago
ByteDance - Software Engineer - Machine Learning Training

ByteDance

Singapore (On-Site)
1 Week ago
Google - Software Engineer III, Generative AI

Google

Sunnyvale, California, United States (On-Site)
1 Week ago
Google - Software Engineer, Black Community Inclusion

Google

(On-Site)
5 Months ago
Trackman - Team Lead - Radar & High-Speed Electronics

Trackman

(On-Site)
4 Weeks ago
ByteDance - Software Engineer Graduate (Applied Machine Learning - Enterprise) - 2025 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Microsoft - Research Intern - Sustainability in Total Cost of Ownership

Microsoft

Hillsboro, Oregon, United States (On-Site)
6 Days ago
Meta - Software Engineer, Machine Learning

Meta

London, England, United Kingdom (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Ubisoft - Senior ML Ops - Content Creation Technology Group

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago
Google - Software Engineer III, Machine Learning, Search

Google

Seattle, Washington, United States (On-Site)
5 Months ago
Dolby Laboratories - AIOps Research Scientist

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
Truecaller - Senior MLOps Engineer

Truecaller

Stockholm, Stockholm County, Sweden (On-Site)
5 Months ago
NVIDIA - Principal Engineer

NVIDIA

(Remote)
1 Month ago
Bragg - AI/ML Engineer

Bragg

Ljubljana, Ljubljana, Slovenia (Hybrid)
1 Week ago
Google - Accelerator Architect and Performance Engineer, Generative AI

Google

San Diego, California, United States (On-Site)
4 Days ago
NVIDIA - Senior AI-HPC Cluster Engineer

NVIDIA

Santa Clara, California, United States (Hybrid)
4 Weeks ago
ByteDance - Software Engineer - Machine Learning Training

ByteDance

Singapore (On-Site)
1 Week ago
Talentica Software - Data Scientist

Talentica Software

India (Remote)
6 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

DraftKings - Senior Product Manager, Live Betting

DraftKings

Boston, Massachusetts, United States (On-Site)
2 Weeks ago
Zoox - Senior/Staff Software Engineer - Simulation Traffic & Behavior Modeling

Zoox

Foster City, California, United States (Hybrid)
6 Months ago
Inkittt - PR Manager

Inkittt

San Francisco, California, United States (On-Site)
10 Months ago
PlayStation Global - Technical Product Manager II

PlayStation Global

Los Angeles, California, United States (Hybrid)
1 Week ago
Netflix - Screen Credits Coordinator, Film

Netflix

Los Angeles, California, United States (On-Site)
5 Days ago
IGN - Lead Product Manager

IGN

New York, New York, United States (Remote)
4 Weeks ago
Activision - Expert Cloud Security Engineer

Activision

United States (On-Site)
2 Weeks ago
Meta - Software Engineer, iOS

Meta

Seattle, Washington, United States (On-Site)
5 Months ago
On Location - Senior Finance Systems Manager

On Location

New York, New York, United States (On-Site)
1 Month ago
ByteDance - Backend Software Engineer - Global E-Commerce Supply Chain Merchant Platform

ByteDance

San Jose, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Google - Software Engineer, Smart Wearable, Google Pixel

Google

New Taipei, New Taipei City, Taiwan (On-Site)
1 Week ago
Ubisoft - Lead R&D Programmer - La Forge

Ubisoft

Montreal, Quebec, Canada (Hybrid)
1 Week ago
Rivos - SOC Static Timing Analysis Engineer - Full Time

Rivos

Hsinchu, Hsinchu City, Taiwan (On-Site)
6 Months ago
ByteDance - Senior Research Scientist- Foundation Model, Vision and Language

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
NVIDIA - Senior Chip Design Engineer

NVIDIA

Yokne'am Illit, North District, Israel (On-Site)
3 Months ago
Google - Senior Physical Design CAD Manager

Google

Mountain View, California, United States (On-Site)
1 Week ago
ByteDance - Research Scientist Graduate (High-Performance Computing (Inference Optimization) - Vision AI Platform)

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
NVIDIA - Backend Engineer, Full Chip Layout

NVIDIA

Iași, Iași County, Romania (Remote)
1 Month ago
NVIDIA - Firmware PHY Verification Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (Hybrid)
3 Months ago
Google - ASIC DFT Engineer, Silicon

Google

Bengaluru, Karnataka, India (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Yokne'am Illit, North District, Israel (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug