NVIDIA Corporation

Senior Machine Learning Engineer, Accelerated Apache Spark

NVIDIA Corporation$224K — $356K *
Information Technology
11 - 15 years of experience
Job Overview by Ladders

Qualifications

  • BS, MS, or PhD in Machine Learning, Data Science, Computer Science or related field.
  • 12+ years of experience in ML/DL solutions.
  • 3+ years as technical lead in ML model development.
  • 2+ years' experience with Apache Spark for large-scale data processing.
  • Strong programming skills in Python and data science libraries (e.g., NumPy, Pandas).
  • Deep experience in advanced ML methodologies like LLM/GenAI and reinforcement learning.
  • Expertise in feature engineering and developing boosted tree models.

Responsibilities

  • Design and implement ML solutions for optimizing GPU-accelerated Apache Spark workloads.
  • Develop advanced algorithms to enhance the performance of Spark on GPUs.
  • Create AI-based tools for system troubleshooting and application optimization.
  • Collaborate with partners on deploying complex ML solutions across environments.
  • Stay updated with the latest developments in ML systems and algorithms.
  • Mentor and lead a team in data science and ML practices.

Benefits

  • Equity opportunities.
  • Comprehensive health benefits.
  • Support for continuous learning and development.
  • Flexible work arrangements to promote work-life balance.
Full Job Description
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. You will work with the open source community to accelerate Apache Spark with GPUs. You will apply the latest ML/AI methods to empower enterprises to migrate Spark workloads onto GPUs at scale.

What you'll be doing:
  • 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.
  • 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).


Ways to stand out from the crowd:
  • 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++.


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.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until September 5, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

About NVIDIA Corporation

Nvidia, a global leader in graphics, gaming, and AI technology, offers Nvidia careers and internship opportunities for those passionate about driving innovation in the tech industry. you'll find a company committed to growth, teamwork, and leadership in computer science and machine learning domains.

About Nvidia

A Pioneer in Technology and Innovation

Nvidia has cemented its reputation as a powerhouse in developing advanced graphics processing units (GPUs) and has significantly contributed to the gaming industry's evolution. Moreover, its foray into AI and machine learning has opened new frontiers in technology, making Nvidia a beacon of innovation and a desirable workplace for ambitious tech professionals.

Job Opportunities

Diverse Positions in a Dynamic Field

Nvidia is continuously on the lookout for talented individuals across various domains, including hardware and software engineering, product design, marketing, and sales. Employment opportunities at Nvidia are vast, catering to a wide range of expertise and career aspirations.

Employment in Hardware and Graphics

For those fascinated by the intricacies of hardware and graphics technology, Nvidia offers positions that sit at the forefront of gaming and computing advancements.

Growth in Machine Learning and AI

Nvidia's leadership in AI and machine learning has created numerous vacancies for specialists eager to contribute to groundbreaking projects.

Recruitment in Computer Science

With the constant demand for innovation, Nvidia's recruitment efforts focus on computer science experts capable of pushing the boundaries of what's possible.

Internship Program

Opening Doors to Future Innovators

Nvidia's internship program is designed to nurture the next generation of technology leaders, offering hands-on experience in a culture that celebrates creativity and teamwork.

Benefits and Culture

Interns at Nvidia enjoy a plethora of benefits, from competitive stipends to mentorship opportunities, all within an environment that values growth and learning.

Opportunities for Students

Whether you're an undergraduate, a master's student, or a Ph.D. candidate, Nvidia's internships provide a real-world glimpse into the tech industry, offering valuable experience in various technology fields.

Pathways to Full-Time Employment

Many interns have transitioned into full-time positions, marking the start of successful careers at Nvidia. The internship program is more than a stepping stone into the company; it’s an investment in the professional development of interns. The goal is to ensure that interns are well-equipped for future challenges.

Nvidia Careers: More Than Just a Job

Nvidia offers more than just a job to its employees; it provides a front-row seat on the journey into the future of technology. Nvidia stands as a pillar of innovation with its vast opportunities in hardware, graphics, gaming, machine learning, and computer science. Nvidia careers serve as a launching pad for talented workers who aim to redefine the technological landscape. Whether through full-time positions or internships, joining Nvidia means contributing to a legacy of breakthroughs and becoming part of a global community dedicated to pushing the boundaries of what's possible.
Learn more about NVIDIA Corporation
Size
22,473 employees
Market Cap
$350.4 billion
Industry
Net Income
$4.3 billion
Founded
1993
5 Year Trend
+31.3%
Revenue
$16.6 billion
NASDAQ

Similar Jobs

More Jobs at NVIDIA Corporation

More Information Technology Jobs

Find similar Senior Machine Learning Engineer, Accelerated Apache Spark jobs: