jobs

Senior Machine Learning Engineer, Accelerated Apache Spark

NVIDIA is looking for a Machine Learning Engineer to join the GPU accelerated Apache Spark team.Apache Spark is the most popular data processing engine in data centers for running massive scale workloads for ETL, SQL, and ML/DL model traini

In inglese

Company

NVIDIA

Location

US, CA, Santa Clara

Status

Open

Posted

2026-09-01T00:00:00+00:00

NVIDIA seeks a machine learning engineer for its GPU-accelerated Apache Spark team in Santa Clara, on-site and full-time. The role involves designing ML solutions for performance prediction and optimization of Spark workloads on GPUs, building adaptive algorithms, and developing AI agents that help fix system issues and tune applications. Candidates need 12+ years of professional ML/DL experience, 3+ years as a technical lead, 2+ years with large-scale data platforms such as Spark, strong Python and libraries like PyTorch and TensorFlow, plus LLM/GenAI, reinforcement learning, and boosted-tree work. Spark internals, CUDA, and Scala/Java/C++ are pluses.