AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

Apple is where individual imaginations gather together, creating innovative products and services that change lives for the better.
$175,800 - $312,200
Machine Learning
Staff Software Engineer
Hybrid
5,000+ Employees
7+ years of experience
AI · Enterprise SaaS

Description For AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

Apple's ML Platform & Technology team is seeking a Staff ML Infrastructure Engineer to join their ML Compute division. This role is crucial in building and maintaining compute infrastructure for ML workloads at Apple, including training, inferencing, and fine-tuning. The position offers an exciting opportunity to tackle complex technical challenges at scale, making ML workloads more stable, reliable, efficient, and cost-effective.

The role combines deep technical expertise with collaborative leadership, requiring both strong engineering skills and the ability to work across teams. You'll be working with cutting-edge ML infrastructure, using technologies like Kubernetes, Python, and Go to build scalable solutions. The position offers competitive compensation ranging from $175,800 to $312,200, along with comprehensive benefits including stock options and educational support.

As a Staff Engineer, you'll be instrumental in architecting and delivering innovative solutions while mentoring team members and driving technical excellence. The role requires 7+ years of experience in building scalable backend systems for ML/DL models, making it perfect for someone who's passionate about machine learning infrastructure and wants to make an impact at Apple's scale.

Apple's commitment to diversity and innovation makes this an ideal opportunity for those looking to work on challenging problems while contributing to products that impact millions of users. The hybrid work environment in Santa Clara offers flexibility while maintaining collaborative opportunities with world-class teams.

Last updated 4 minutes ago

Responsibilities For AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

  • Scope, architect, and deliver innovative high-quality solutions
  • Effectively communicate complex features and systems in detail
  • Collaborate with teams across Apple on ML workloads such training, inferencing, fine tuning etc.
  • Understand industry & company-wide trends to help assess & develop new technologies
  • Code using Go, Python
  • Conduct code reviews
  • Help onboard new team members, provide mentorship

Requirements For AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

Go
Python
Kubernetes
  • Bachelors in Computer Science, engineering, or a related field
  • 7+ years of hands-on experience in building scalable backend systems for training and evaluation of machine learning/deep learning models
  • Proficient in relevant programming languages, like Python or Go
  • Strong expertise in distributed systems, reliability and scalability, containerization, and cloud platforms
  • Proficient in cloud computing and data processing infrastructure and tools: Kubernetes, Ray, PySpark
  • Ability to clearly and concisely communicate technical and architectural problems

Benefits For AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

Medical Insurance
Dental Insurance
Education Budget
Equity
Relocation Benefits
  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Employee stock programs
  • Education reimbursement
  • Discretionary bonuses
  • Relocation assistance
  • Employee discount on products
  • Free services

Interested in this job?

Jobs Related To Apple AIML - Staff ML Infrastructure Engineer, ML Platform & Technology - ML Compute

SWE - Machine Learning Technical Lead, Data Engineering

Lead Machine Learning Technical role at Apple focusing on ML data infrastructure, pipeline development, and team leadership.

AIML - Sr Engineering Program Manager, ML Developer Tools, Metrics, and Infrastructure

Senior Engineering Program Manager position at Apple, focusing on ML Developer Tools and Infrastructure, offering competitive salary and benefits in Seattle.

AIML - Sr Engineering Program Manager, NLU Evaluation

Senior Engineering Program Manager position at Apple, leading NLU technology development for products like Siri, requiring 6+ years of ML product experience.

AIML - Machine Learning Engineer, Foundation Models

Senior ML Engineer role at Apple focusing on building and deploying foundation models for various Apple products.

AIML - Sr Engineering Program Manager, Siri Devices

Senior Engineering Program Manager position at Apple leading Siri Devices initiatives, requiring 5+ years of experience in program management and software engineering.