All vacancies
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
- Senior AI Engineer/ Node.js, Python, TypescriptEPAM Systems · Remote · Armenia
- Machine Learning EngineerYour Personal AI · Remote · worldwide, UTC+4 accepted
- Lead AI Engineer with Machine LearningEPAM Systems · Remote · Armenia
- MLOps Engineer InternshipMetasys · Remote · worldwide, UTC+4 accepted
- Lead Data Engineer- Healthcare InteroperabilityVerisma · Remote · worldwide, UTC+4 accepted
- Data EngineerTechBiz Global · Remote · worldwide, UTC+4 accepted

