Senior Machine Learning Engineer, Accelerated Apache Spark - CA, Santa Clara, United States
$224,000 - $356,500 USD yearly
Originally posted 1 September 2026 by the employer.
About the role
What you'll do
- 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 partners and customers on the deployment of complex machine learning solutions.
- 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.
What you'll need
- 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+ years experience as technical lead in ML model development.
- 2+ years of hands-on experience with large-scale data processing platforms, such as Apache Spark.
- Proven ability to employ modern tooling and sound techniques for crafting, deploying, and maintaining machine learning models.
- Excellent programming skills in Python and Python data science related libraries like numpy, pandas, scikit-learn, scipy, pytorch, and tensorflow.
- Deep experience with sophisticated ML methodologies, including LLM/GenAI, reinforcement learning, and adaptive, on-line ML systems.
- Strong expertise in feature engineering, feature importance assessment, and developing boosted tree model solutions (e.g., XGBoost).
Skills: Machine Learning Engineer, Apache Spark, GPU accelerated, ML/DL solutions, AI-based agents, CUDA
This role has been open 43 days — below the 51-day median for Infrastructure/Platform Software roles.
Infrastructure/Platform Software · Infrastructure Platform
|
Open roles in category
956
|
Median days open
51 d
|
Median salary
$235k
|
See the full market breakdown ▾Category comparison, skills in demand, and who else is hiring
| Metric | NVIDIA | All employers we track in this specialty (956 roles · 86 employers) |
|---|---|---|
| Open roles in this specialty | 289 | 956 |
| Open roles in the wider Software, Firmware & Systems family | 1008 | 6164 · 146 employers |
| Median days open | 43 d | 51 d (−8 d vs this employer) |
| Median salary (USD postings) | — | $235k |
Skills observed across this category: Machine Learning Engineer, Apache Spark, GPU accelerated, ML/DL solutions, AI-based agents, CUDA
Who's hiring in this category
- NVIDIA (this employer) · 287 open roles · median 44 d
- Qualcomm · 61 open roles · median 56 d
- AMD · 53 open roles · median 22 d
- Graphcore · 44 open roles · median 89 d
- Cerebras · 39 open roles · median 79 d
- Intel Corporation · 36 open roles · median 13 d
How we counted: 956 open Infrastructure/Platform Software (Infrastructure Platform) roles from 86 employers tracked in the SemiconductorJobs index, counted 14 Oct 2026. Specialty figures count only roles carrying this exact specialty label, so an employer's related work in neighbouring specialties is not included there — it is counted in the wider Software, Firmware & Systems family row. Figures refresh nightly.