Senior Machine Learning Engineer, Accelerated Apache Spark

NVIDIA
Santa Clara
Full timeUSD 224,000 - 356,500 annuallyFunction: Data Science & Machine LearningExperience: 12+ yearsSkills: ["Technical mentorship","Leadership","Collaboration","Problem solving"]

Design and implement GPU-accelerated machine learning solutions for performance prediction and optimization of Apache Spark workloads. Build adaptive algorithms that continuously improve Spark execution on NVIDIA GPUs, and develop AI agents and tools to help diagnose issues and optimize applications. Collaborate with partners and customers to deploy complex ML solutions across environments while staying current on ML systems advances and providing technical mentorship to a team.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
3 days ago

Senior Machine Learning Engineer, Accelerated Apache Spark

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 59 minutes agoStatus: Live

Job Summary

Design and implement GPU-accelerated machine learning solutions for performance prediction and optimization of Apache Spark workloads. Build adaptive algorithms that continuously improve Spark execution on NVIDIA GPUs, and develop AI agents and tools to help diagnose issues and optimize applications. Collaborate with partners and customers to deploy complex ML solutions across environments while staying current on ML systems advances and providing technical mentorship to a team.
Location: Santa Clara
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Design and implement ML solutions to predict and optimize performance of GPU-accelerated enterprise Apache Spark workloads.
  • •Develop advanced algorithms and adaptive systems to continuously improve Spark workload performance on GPUs.
  • •Build AI-based agents and tools to assist with fixing system issues and application optimization.
  • •Collaborate with partners and customers to deploy complex ML solutions in different environments.
  • •Provide technical mentorship and leadership to a team of engineers in data science and machine learning.

Pay and Benefits

Salary: USD 224,000 - 356,500 annually
Equity and Bonus:Equity

Key Requirements

  • •BS, MS, or PhD (or equivalent) in Machine Learning, Data Science, Computer Science, or a related field.
  • •12+ years designing, implementing, and productionizing high-quality ML/DL solutions.
  • •3+ years of experience as a technical lead in ML model development.
  • •2+ years hands-on experience with large-scale data processing platforms such as Apache Spark.
  • •Strong programming and ML experience in Python and libraries such as numpy, pandas, scikit-learn, scipy, PyTorch, and TensorFlow, plus advanced ML methods (e.g., LLM/GenAI, reinforcement learning).
Experience:12+ yearsEnterpriseOpen sourceData processingGPUML/DLGenAI
Education:
Skills:Technical mentorshipLeadershipCollaborationProblem solving
Tech Stack:Machine LearningDeep LearningApache SparkGPUNVIDIA GPUsCUDAPythonNumpyPandasScikit-learnScipyPyTorchTensorFlowLLMGenAIReinforcement learningOnline ML systemsFeature engineeringXGBoostScala

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor