Nvidia
Senior Machine Learning Engineer, Accelerated Apache Spark
US, CA, Santa Clara · Senior
Stay score
odds of building a lasting career here
Thin sponsorship signal and lottery-bound. A low-probability bet with your clock running. Prioritize cap-exempt roles and proven entry-level sponsors first.
Lottery odds assume a STEM candidate.
Personalize to your clock →Employer immigration record
from this employer's Department of Labor filings
Green-card filing pattern in this occupation
Green-card intent detected
Green-card follow-through: 92%
Files H-1B transfers
Sourced from Department of Labor LCA, PERM and prevailing-wage disclosure data. Employer matching is by name, so figures may be split across an employer's legal entities. Absence of a filing means none appears in our copy of the data, not that none exists.
Community outcomes
No reports yet — be the first to help the next applicant.
About the role
- NVIDIA is looking for a Machine Learning Engineer to join the GPU accelerated Apache Spark team.
- 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
- 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.
- Maintain deep domain expertise by knowing the latest published advances in ML systems and algorithms.
- 3+ experience as technical lead in ML model development.
- Strong expertise in feature engineering, feature importance assessment, and developing boosted tree model solutions (e.g., XGBoost).
Requirements
- If you are passionate about what you do, creative and autonomous, we want to hear from you!
Compensation
- Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
- The base salary range is 224,000 USD - 356,500 USD.
Benefits
- Design and implement machine learning solutions for performance prediction and optimization of GPU accelerated enterprise Apache Spark workloads.
- Collaborate with key partners and customers on the deployment of complex machine learning solutions in various environments.
- Provide technical mentorship and leadership in data science and machine learning to a team of engineers.
- BS, MS, or PhD or equivalent experience in Machine Learning, Data Science, Computer Science or a closely related field.
- Proven ability to employ modern tooling and sound techniques for all aspects of crafting, deploying, and maintaining machine learning models.
- Deep experience with sophisticated ML methodologies, including LLM/GenAI, reinforcement learning, and adaptive, on-line ML systems.
Company info
- Ways to stand out from the crowd: Understanding of the internal workings and architecture related to Apache Spark.
Equal opportunity
- NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer.
This listing is sourced directly from Nvidia's careers page and normalized into a canonical job model.