# Senior Machine Learning Engineer, Accelerated Apache Spark

**Company**: NVIDIA
**Location**: Santa Clara, CA
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Machine-Learning-Engineer--Accelerated-Apache-Spark_JR2024455?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_31fe845f-a56

## Description

NVIDIA is looking for a Senior Machine Learning Engineer to join the GPU accelerated Apache Spark team.

You will work on designing and implementing machine learning solutions for performance prediction and optimization of GPU accelerated enterprise Apache Spark workloads.

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.

Requirements:

- 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).

Benefits:

- Equity

- Benefits

## Skills

### Required
- Machine Learning
- Data Science
- Computer Science
- Apache Spark
- Python
- numpy
- pandas
- scikit-learn
- scipy
- pytorch
- tensorflow
- LLM/GenAI
- reinforcement learning
- adaptive ML systems
- feature engineering
- XGBoost

### Nice to have
- Understanding of the internal workings and architecture related to Apache Spark
- Familiarity with NVIDIA GPUs and CUDA
- Experience coding in Scala, Java, and/or C++

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-Machine-Learning-Engineer--Accelerated-Apache-Spark_JR2024455?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)