Senior ML Engineer - GPU-Accelerated Spark

NVIDIA; Santa Clara (CA); On-site; USD 224,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team, focusing on performance prediction and optimization of GPU accelerated enterprise Apache Spark workloads. You will apply the latest ML/AI methods, mentor engineers, and collaborate with partners on deployment across environments, while staying current with advances in ML systems.

The base salary range is 224,000 USD - 356,500 USD, with equity and benefits offered.

Qualifications

• 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.
• Proven hands‑on experience (2+ years) with large-scale data processing platforms, such as Apache Spark.
• Proven ability to employ modern tooling and sound techniques for all aspects of 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).

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.

Skills

Python

Spark

ML/DL

GenAI

Reinforcement Learning

Feature Engineering

XGBoost

Leadership

Education

BS/MS/PhD in ML/DS/CS

Job description

NVIDIA is seeking a Machine Learning Engineer to join the GPU accelerated Apache Spark team, focusing on performance prediction and optimization of GPU accelerated enterprise Apache Spark workloads. You will apply the latest ML/AI methods, mentor engineers, and collaborate with partners on deployment across environments, while staying current with advances in ML systems.

The base salary range is 224,000 USD - 356,500 USD, with equity and benefits offered.