Principal Machine Learning Engineer, Accelerated Apache Spark

NVIDIA • Santa Clara, CA, United States • Posted June 06, 2026

Location Santa Clara, CA
Job Type Full-time
Category other-general
Posted June 06, 2026
NVIDIA is looking for a Machine Learning (ML) Engineer to join the GPU accelerated Apache Spark team. Apache Spark is the most popular data processing engine in data centers for running large scale workloads for ETL, SQL, and ML/DL model training and inference pipelines, spanning many domains and use cases. NVIDIA GPUs offer a promising avenue for significantly speeding up and/or lowering the cost of running Apache Spark applications at massive scales. You will work with the open source community to accelerate Apache Spark with GPUs. You will apply the latest ML/AI methods to empower enterprises to migrate Spark workloads onto GPUs at scale.


What you’ll be doing:
+ Design and implement machine learning solutions for performance prediction and optimization of GPU accelerated enterprise Apache Spark workloads.
+ Develop advanced algorithms and adaptive systems to continuously improve the performance of Apache Spark workloads on GPUs.
+ Develop AI-based agents and ...

Interested in this role?

Click the button below to start your application.

Apply Now