Senior Math Libraries Engineers - Python APIs

5 Months ago • 3 Years + • Programming • $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

NVIDIA - System Debug Lead Engineer

NVIDIA

Santa Clara, California, United States (On-Site)
3 Months ago
Qualcomm - STCO PDK Development and Custom Layout Integration Engineer

Qualcomm

Hsinchu City, Taiwan (On-Site)
2 Months ago
Enphase Energy - Web Performance & Analytics Specialist

Enphase Energy

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Figma - People Relations Partner

Figma

London, England, United Kingdom (Remote)
1 Month ago
CME Group - Lead Analyst, Value Management

CME Group

Bengaluru, Karnataka, India (On-Site)
1 Year ago
Roof Stacks - Software Developer

Roof Stacks

Istanbul, İstanbul, Türkiye (On-Site)
4 Months ago
Ubisoft - Junior Programmer

Ubisoft

Chengdu, Sichuan, China (On-Site)
3 Months ago
Electronic Arts - Software Developer II - AdTech

Electronic Arts

Vancouver, British Columbia, Canada (Hybrid)
3 Months ago
Capgemini - Linux Administrator with SQL

Capgemini

Mumbai, Maharashtra, India (On-Site)
2 Months ago
5minlab - Client Programmer

5minlab

Seoul, South Korea (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

SciPlay - Senior Cloud Engineer

SciPlay

Austin, Texas, United States (Hybrid)
5 Months ago
White board games - 3D Character Artist

White board games

Argentina (Remote)
3 Months ago
Qualcomm - Sr Lead Engineer- Display

Qualcomm

Hyderabad, Telangana, India (On-Site)
2 Months ago
Garena - Associate/Senior Associate, Revenue Operations

Garena

Bengaluru, Karnataka, India (On-Site)
3 Months ago
SEGA - Senior/Principal Graphics Programmer

SEGA

England, United Kingdom (On-Site)
4 Months ago
HCL Tech - Administrator

HCL Tech

Amsterdam, North Holland, Netherlands (On-Site)
1 Month ago
White board games - QA Analyst (SSR)

White board games

Argentina (Remote)
3 Months ago
Infosys - Senior Developer - NextJS Full Stack (with Sitecore/CMS Exposure)

Infosys

Melbourne, Victoria, Australia (On-Site)
3 Months ago
Jumio - Customer Escalation Engineer

Jumio

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Mapbox - Software Development Engineer II, 3D Navigation, C++

Mapbox

Poland (Remote)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Oregon, Remote, United States

FICO - Principal Cloud Engineer

FICO

United States (Remote)
1 Year ago
Yodlee - Business Assurance Security Manager

Yodlee

Raleigh, North Carolina, United States (Hybrid)
1 Month ago
Litmus - Partner Sales Engineer – Cloud & Industrial Analytics

Litmus

United States (Remote)
1 Month ago
London stock Exchange - Major Incident Manager

London stock Exchange

St. Louis, Missouri, United States (On-Site)
2 Months ago
Axon - Senior Manager, Billing Compliance

Axon

Boston, Massachusetts, United States (Hybrid)
2 Months ago
Kavalirio - Manufacturing Engineer Level 4

Kavalirio

Los Angeles, California, United States (On-Site)
3 Months ago
Apple - Budget Manager, PD&M Operations

Apple

Culver City, California, United States (On-Site)
1 Month ago
Aptive - Executive Assistant to the Executive VP and Chief Financial Officer

Aptive

Boston, Massachusetts, United States (On-Site)
3 Months ago
UPF Industries  - Design Supervisor

UPF Industries

San Antonio, Texas, United States (On-Site)
2 Months ago
Apple - Data Engineer, PART DataOps

Apple

Cupertino, California, United States (On-Site)
4 Weeks ago

Get notifed when new similar jobs are uploaded

Programming Jobs

Capgemini - Azure .Net Developer

Capgemini

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Ion - Java Developer, Budapest

Ion

Budapest, Hungary (Hybrid)
9 Months ago
PayPal - Database Engineer

PayPal

Singapore (Hybrid)
1 Month ago
Tesla - Senior PLC Programmer

Tesla

Prüm, Rhineland-Palatinate, Germany (On-Site)
5 Months ago
BioFire - Manufacturing Technician - (C Shift, Nights)

BioFire

Salt Lake City, Utah, United States (On-Site)
3 Months ago
Capgemini - C++ Dev

Capgemini

Pune, Maharashtra, India (On-Site)
3 Months ago
plana technologies - SQL Developer

plana technologies

Hyderabad, Telangana, India (Remote)
3 Weeks ago
P99 soft - Sr Java & Python Developer

P99 soft

Hyderabad, Telangana, India (On-Site)
3 Months ago
Nagarro - Associate Staff Engineer, Java

Nagarro

Philippines (On-Site)
9 Months ago
Mcdonalds - Software Engineer III - Java

Mcdonalds

Mexico City, Mexico (Hybrid)
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.

Taipei City, Taiwan (On-Site)

Beijing, Beijing, China (On-Site)

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

Santa Clara, California, United States (Hybrid)

Bengaluru, Karnataka, India (Hybrid)

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

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

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

Dubai, Dubai, United Arab Emirates (On-Site)

Beijing, Beijing, China (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