Full-Time

Principal Engineer

Distributed Machine Learning

Confirmed live in the last 24 hours

NVIDIA

NVIDIA

10,001+ employees

Designs GPUs and AI computing solutions

Automotive & Transportation
Enterprise Software
AI & Machine Learning
Gaming

Compensation Overview

$272k - $425.5kAnnually

+ Equity

Expert

Santa Clara, CA, USA

Category
Applied Machine Learning
Deep Learning
AI & Machine Learning
Software Engineering
Required Skills
Kubernetes
Python
Pytorch
Apache Spark
Scala
C/C++
Yarn
Requirements
  • BS, MS, or PhD in Computer Science, Computer Engineering, or closely related field (or equivalent experience).
  • 12+ years of work or research experience in software development.
  • 5+ experience as technical lead in distributed machine learning and/or deep learning.
  • 3+ years of open source development experience.
  • 3+ years of hands-on experience with Spark MLlib, XGBoost, and/or PyTorch.
  • Knowledge of internals of Apache Spark MLlib.
  • Experience with Kubernetes, YARN, Spark, and/or Ray for distributed ML orchestration.
  • Proven technical skills in designing, implementing and delivering high-quality distributed systems.
  • Excellent programming skills in C++, Scala, and Python.
  • Familiar with agile software development practice.
Responsibilities
  • Design and develop new user-friendly APIs and libraries to optimally use existing DL/ML frameworks in GPU-enabled Spark clusters for distributed DL/ML training and inference at scale.
  • Design and develop GPU accelerated ML libraries for distributed training and inference on Spark clusters, e.g., to improve our existing spark-rapids-ml open source library.
  • Demonstrate superior performance of developed solutions on industry standard benchmarks and datasets.
  • Make technical contributions to enhance capabilities of open source projects such as RAPIDS, XGBoost, spark-rapids-ml, and Apache Spark.
  • Work with NVIDIA partners and customers on deploying distributed ML algorithms in cloud or on-premise.
  • Keep up with published advances in distributed ML systems and algorithms.
  • Provide technical mentorship to a team of engineers.

NVIDIA designs and manufactures graphics processing units (GPUs) and system on a chip units (SoCs) for various markets, including gaming, professional visualization, data centers, and automotive. Their main products are GPUs that enhance gaming experiences and support professional applications, along with AI and high-performance computing platforms tailored for developers and data scientists. NVIDIA stands out from competitors by offering a combination of hardware and software solutions, including cloud-based services like NVIDIA CloudXR and NGC, which enable scalable applications in AI and machine learning. The company's goal is to drive innovation in technology and provide advanced solutions that cater to a wide range of clients, from gamers to enterprises.

Company Stage

IPO

Total Funding

$19.5M

Headquarters

Santa Clara, California

Founded

1993

Growth & Insights
Headcount

6 month growth

2%

1 year growth

1%

2 year growth

0%
Simplify Jobs

Simplify's Take

What believers are saying

  • Acquisition of VinBrain enhances NVIDIA's AI capabilities in the healthcare sector.
  • Investment in Nebius Group boosts NVIDIA's AI infrastructure and cloud platform offerings.
  • Serve Robotics' expansion, backed by NVIDIA, highlights growth in autonomous delivery services.

What critics are saying

  • Increased competition from AI startups like xAI could challenge NVIDIA's market position.
  • Serve Robotics' expansion may divert resources from NVIDIA's core GPU and AI businesses.
  • Integration of VinBrain may pose challenges and distract from NVIDIA's primary operations.

What makes NVIDIA unique

  • NVIDIA leads in AI and HPC solutions with cutting-edge GPU technology.
  • The company excels in diverse markets, including gaming, data centers, and autonomous vehicles.
  • NVIDIA's cloud services, like CloudXR, offer scalable solutions for AI and machine learning.

Help us improve and share your feedback! Did you find this helpful?

Benefits

Company Equity

401(k) Company Match