Skip to content
ITA Jobs
All vacancies
NVIDIA logo

Senior Machine Learning Engineer, Accelerated Apache Spark

NVIDIA

US, CA, Santa Clara · On-site · relocation$18,667–29,708 gross/mofull-timeNVIDIA Careers (offices abroad)

Apache Spark is the most popular data processing engine in data centers for running massive 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 very large Apache Spark applications.

Responsibilities

  • 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 tools to assist with fixing system issues and application optimization.
  • Collaborate with key partners and customers on the deployment of complex machine learning solutions in various environments.
  • Maintain deep domain expertise by knowing the latest published advances in ML systems and algorithms.
  • Provide technical mentorship and leadership in data science and machine learning to a team of engineers.
  • What we need to see
  • BS, MS, or PhD or equivalent experience in Machine Learning, Data Science, Computer Science or a closely related field.
  • 12+ years of professional experience in designing, implementing, and productionizing high-quality ML/DL solutions.
  • 3+ experience as technical lead in ML model development.

Languages

    Work format
    On-site
    Seniority
    Senior
    Posted
    1 Sept 2026
    Last verified
    In the last 5 days

    Similar roles

    Keep looking