Senior Math Libraries Engineers - Python APIs

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

Job Summary

Job Description

NVIDIA seeks a Senior Math Libraries Engineer to design and develop high-performance Python APIs for their math libraries. This role involves crafting and developing Python APIs (e.g., linear algebra), integrating them into libraries and frameworks, conducting performance analysis and benchmarking, and optimizing numerical software on CPU and GPU architectures. Collaboration with product management and partners is key to understanding feature requirements and contributing to technical roadmaps. The ideal candidate possesses excellent Python and C++ skills, experience with numerical methods, and a strong understanding of parallel computing.
Must have:
  • PhD or MSc in relevant field or equivalent experience
  • Excellent Python and C++ skills
  • Experience developing Python APIs
  • 3+ years experience in high-performance application development
  • Strong understanding of numerical methods
Good to have:
  • GPU programming experience (CUDA or OpenCL)
  • Knowledge of CPU/GPU architecture
  • Experience with NumPy, CuPy, SciPy, JAX, PyTorch, TensorFlow, or Numba
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

Google - Machine Learning Algorithm Engineer, Silicon

Google

Mountain View, California, United States (On-Site)
1 Month ago
Qualcomm - Machine Learning & AI ADAS Applications Senior Staff Engineer

Qualcomm

Shanghai, China (On-Site)
2 Weeks ago
Scale AI - SEAL Research Scientist/ Research Engineer

Scale AI

San Francisco, California, United States (On-Site)
1 Month ago
attentive - Senior Machine Learning Engineer

attentive

San Francisco, California, United States (Hybrid)
7 Months ago
Moloco - Staff Machine Learning Engineer

Moloco

Seoul, South Korea (On-Site)
2 Weeks ago
Google - Staff Software Engineer, Infrastructure, Platforms Infrastructure Engineering

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
NVIDIA - Senior Optical MSDV Hardware Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
4 Months ago
NVIDIA - Manager, Software Engineering

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
NVIDIA - Deep Learning Performance Architect

NVIDIA

Beijing, Beijing, China (On-Site)
3 Months ago
NVIDIA - Senior STA Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Krafton - Deep Learning Engineer - RL

Krafton

Seoul, South Korea (On-Site)
4 Months ago
ManyChat - Lead Machine Learning Scientist

ManyChat

Amsterdam, North Holland, Netherlands (Hybrid)
3 Days ago
Balbix - AI/ML Architect

Balbix

Bengaluru, Karnataka, India (On-Site)
7 Months ago
NVIDIA - Senior Research Scientist, Multimodal Foundation Models and Robotics

NVIDIA

Santa Clara, California, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, Distributed Machine Learning

Google

Mountain View, California, United States (On-Site)
1 Month ago
PlayStation Global - Senior Software Engineer (Back End / Machine Learning)

PlayStation Global

San Mateo, California, United States (On-Site)
1 Month ago
Ciklum - Expert Data Scientist

Ciklum

Pune, Maharashtra, India (Hybrid)
7 Months ago
London stock Exchange - Senior AI Platform Engineer

London stock Exchange

London, England, United Kingdom (On-Site)
1 Week ago
Canva - Senior Applied Scientist - AI Research

Canva

Surry Hills, New South Wales, Australia (Remote)
2 Months ago
Razer - AI Software Engineer

Razer

Singapore (On-Site)
3 Days ago

Get notifed when new similar jobs are uploaded

Jobs in Oregon, Remote, United States

Nintendo - CONTRACT - Localization Specialist (Japanese)

Nintendo

Redmond, Washington, United States (Hybrid)
6 Months ago
Netflix - Data Analyst, Production Finance Operations & Innovation

Netflix

Los Angeles, California, United States (On-Site)
1 Month ago
Playstation - IT Support Specialist

Playstation

Los Angeles, California, United States (On-Site)
2 Weeks ago
UPF Industries  - Maintenance Technician

UPF Industries

Lansing, Michigan, United States (On-Site)
2 Weeks ago
Trek - Future Store Manager

Trek

Tacoma, Washington, United States (On-Site)
3 Months ago
2K - Senior Release Manager

2K

Las Vegas, Nevada, United States (On-Site)
5 Months ago
2K - Release Technician

2K

Las Vegas, Nevada, United States (On-Site)
2 Months ago
Ethernovia - Senior Field Application Engineer Lead

Ethernovia

San Jose, California, United States (On-Site)
2 Weeks ago
Whatnot - Senior FP&A Analyst

Whatnot

Los Angeles, California, United States (On-Site)
1 Month ago
Riot Games - Manager, Software Engineering - Payments

Riot Games

Los Angeles, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Ubisoft - Lead R&D Scientist

Ubisoft

Shanghai, Shanghai, China (On-Site)
1 Month ago
Ubisoft - Senior C++ Programmer

Ubisoft

Malmö, Skåne County, Sweden (Hybrid)
1 Month ago
bytedance - GPU/AI Application Platform Engineer Intern (Server Platform)

bytedance

San Jose, California, United States (On-Site)
3 Months ago
Krafton - PUBG IP Franchise China Strategic Business PM (3+ years)

Krafton

Seoul, South Korea (On-Site)
1 Month ago
NVIDIA - Senior STA Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
Riot Games - Principal Software Engineer, Product Tech-Lead - Unpublished R&D Product

Riot Games

Los Angeles, California, United States (On-Site)
6 Months ago
NVIDIA - Manager, Software Engineering

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 Months ago
NVIDIA - Senior Mixed-Signal Design Verification Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
rivos - CPU Design Verification - Full-time

rivos

Hsinchu, Hsinchu City, Taiwan (Hybrid)
7 Months ago
bytedance - Principal Algorithm Engineer, Trust and Safety

bytedance

Singapore (On-Site)
7 Months 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)

Massachusetts, United States (On-Site)

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

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

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

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Santa Clara, California, United States (Hybrid)

Pune, Maharashtra, India (On-Site)

Taipei City, Taiwan (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