π Local Job Near You
Principal Machine Learning Engineer, Accelerated Apache Spark
NVIDIA
π
Santa Clara, United States
Location
Santa Clara
Posted
June 03, 2026
Commute
Local Area
Local Opportunity Near You!
This job is in your area. Enjoy a short commute and work close to home.
Job Description
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 ...
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 ...