Principal ML Engineer - GPU-Accelerated Spark + Equity

NVIDIA; Santa Clara (CA); On-site; USD 272,000 - 431,250; Full time

Job summary

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team in Santa Clara, California. You will design and implement innovative machine learning solutions, optimize workloads on GPUs, and work closely with the open-source community.

The ideal candidate should have over 12 years of professional experience in ML/DL solutions, strong programming skills in Python, and a solid understanding of data processing platforms like Apache Spark. The role offers a highly competitive salary range of $272,000 to $431,250.

Qualifications

• 12+ years of professional experience in designing and productionizing high-quality ML/DL solutions.
• 5+ years of experience as technical lead in ML model development.
• Proven hands-on experience with large-scale data processing platforms like Apache Spark.

Responsibilities

• Design and implement ML solutions for GPU accelerated Apache Spark workloads.
• Develop algorithms to continuously improve performance of workloads on GPUs.
• Collaborate with partners on deploying complex ML solutions.

Skills

Machine Learning

Data Science

Python

Apache Spark

Machine Learning Model Development

CUDA

Education

BS, MS, or PhD in Machine Learning, Data Science, Computer Science or related field

Tools

TensorFlow

PyTorch

NumPy

Pandas

Scikit-learn

SciPy

XGBoost

Job description

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team in Santa Clara, California. You will design and implement innovative machine learning solutions, optimize workloads on GPUs, and work closely with the open-source community.

The ideal candidate should have over 12 years of professional experience in ML/DL solutions, strong programming skills in Python, and a solid understanding of data processing platforms like Apache Spark. The role offers a highly competitive salary range of $272,000 to $431,250.