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. Responsibilities include crafting and developing Python APIs (e.g., linear algebra), integrating APIs into libraries/frameworks, performance analysis and tuning, collaborating with product management, and improving user experience. The ideal candidate possesses strong Python and C++ skills, experience with numerical methods, and expertise in high-performance computing. Experience with NumPy, CuPy, SciPy, JAX, PyTorch, TensorFlow, or Numba is a plus. The role requires working in an agile environment and strong communication skills.
Must have:
  • PhD or MSc in CS/Math or equiv. experience
  • Excellent Python & C++ skills
  • Experience developing Python APIs
  • High-performance computing experience (3+ years)
  • Strong numerical methods understanding
Good to have:
  • GPU programming (CUDA/OpenCL)
  • CPU/GPU architecture knowledge
  • Experience with NumPy, CuPy, SciPy, JAX, PyTorch, TensorFlow, 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

IMC - Machine Learning Engineer

IMC

Amsterdam, North Holland, Netherlands (On-Site)
1 Month ago
bytedance - Research Scientist Intern (Doubao (Seed) - Foundation Model, Speech Understanding) - 2024 Summer (PhD)

bytedance

San Jose, California, United States (On-Site)
7 Months ago
Meta - Software Engineer, Machine Learning

Meta

Los Angeles, California, United States (On-Site)
7 Months ago
Google - Software Engineer III, Generative AI

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Krafton - Lead of Physical AI Agent, Research Scientist

Krafton

Seoul, South Korea (On-Site)
1 Month ago
Info Stretch - Engineering Lead - ETL Architecture

Info Stretch

Bengaluru, Karnataka, India (On-Site)
7 Months ago
Meta - Software Engineer, Machine Learning

Meta

Burlingame, California, United States (On-Site)
7 Months ago
NVIDIA - Senior ASIC Verification Engineer, Coherent High Speed Interconnect

NVIDIA

Santa Clara, California, United States (Hybrid)
2 Months ago
bytedance - Research Scientist/Engineer - Multimodal Interaction & World Model

bytedance

Singapore (On-Site)
6 Months ago
CLO Virtual Fashion  Inc  - C++ Developer

CLO Virtual Fashion Inc

Bengaluru, Karnataka, India (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Razer - Solutions Architect

Razer

Singapore (On-Site)
8 Months ago
Cerence - Principal Machine Learning Operations Developer - Inference Optimization

Cerence

Montreal, Quebec, Canada (Hybrid)
1 Month ago
bytedance - Software Engineer in ML Engineering Platform

bytedance

Seattle, Washington, United States (On-Site)
7 Months ago
Ubisoft - Lead R&D Scientist

Ubisoft

Shanghai, Shanghai, China (On-Site)
4 Months ago
The Walt Disney Company - Lead Software Engineer, Machine Learning - Ad Platforms

The Walt Disney Company

San Francisco, California, United States (On-Site)
6 Months ago
Reddit - Senior Machine Learning Engineer

Reddit

United Kingdom (Remote)
1 Week ago
Whatnot - Machine Learning Engineer

Whatnot

Los Angeles, California, United States (On-Site)
2 Weeks ago
Kwalee - Machine Learning Engineer

Kwalee

Royal Leamington Spa, England, United Kingdom (On-Site)
3 Months ago
Match Group - Sr. Software Engineer, Generative AI

Match Group

Palo Alto, California, United States (Hybrid)
7 Months ago
Tencent - Senior Researcher: Artificial General Intelligence (Natural Language Processing)

Tencent

Washington, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Louisiana, United States

bytedance - Enterprise Security Risk Manager

bytedance

New York, New York, United States (On-Site)
1 Month ago
Interactive Brokers - Tax Manager – International Tax Compliance

Interactive Brokers

Greenwich, Connecticut, United States (Hybrid)
3 Weeks ago
Nintendo - Account Planner - Publisher and Developer Relations

Nintendo

Redmond, Washington, United States (On-Site)
2 Months ago
Sloyd - Digital Content Marketing Intern

Sloyd

Austin, Texas, United States (Hybrid)
1 Week ago
smarsh - Future Opportunity - Enterprise Customer Success Manager

smarsh

New York, United States (Hybrid)
1 Month ago
cyara - Account Executives (East)

cyara

United States (Remote)
6 Months ago
Flow - Senior Technical Program Manager

Flow

Palo Alto, California, United States (Hybrid)
1 Month ago
Next Level Business Services - PL/SQL Developer

Next Level Business Services

Buffalo Grove, Illinois, United States (On-Site)
7 Months ago
Cognite - Senior Front-end Engineer - Canvas

Cognite

Phoenix, Arizona, United States (Hybrid)
4 Months ago
Marvell - Test Operations Senior Test Equipment Manager

Marvell

Santa Clara, California, United States (On-Site)
6 Days ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Tesla - Senior PLC Programmer

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
3 Months ago
bytedance - Research Scientist Graduate (Foundation Model, Video Generation) - 2025 Start (PhD)

bytedance

Seattle, Washington, United States (On-Site)
7 Months ago
Krafton - Game Security Product Manager

Krafton

Seoul, South Korea (On-Site)
2 Months ago
Microsoft - Senior Machine Learning Engineer

Microsoft

Vancouver, British Columbia, Canada (On-Site)
1 Month ago
NVIDIA - Senior System Software Architect, HPC Networking

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
3 Months ago
NVIDIA - Developer Technology Engineer - HPC and AI

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
NVIDIA - Mixed-Signal Circuit Design Engineer - New College Graduate

NVIDIA

Taipei City, Taiwan (On-Site)
3 Months ago
NVIDIA - VLSI Physical Design Engineer - New College Grad 2025

NVIDIA

Massachusetts, United States (On-Site)
1 Month ago
NVIDIA - STA Engineer

NVIDIA

Tel Aviv-Yafo, Tel Aviv District, Israel (On-Site)
4 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