← Back to opportunities

Principal Machine Learning Engineer, Accelerated Apache Spark

📍 Location
Santa Clara
⏰ Job Type
Full-time
📅 Posted
June 06, 2026

About the Role

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 ...

Ready to Join Through a Referral?

Apply now and get connected directly with the hiring team

Apply for this Position