Staff Machine Learning Engineer, Analytics & Data Engineering

What you'll do

• Research and develop AI/ML solutions and tools for model training, deployment, and monitoring.
• Design secure, private, high-performance systems and collaborate across Apple as a technical leader, including mentoring junior engineers.
• Improve engineering quality through design and code reviews, test automation, and resilient, maintainable systems.

What they ask for

• PhD or Master’s degree in Machine Learning, Computer Science, Computer Engineering, or equivalent practical experience.
• 5 - 7+ years of professional software engineering experience focused on AI/ML engineering.
• Expert proficiency in Python and/or Java/Scala, with strong computer science fundamentals and system design skills.
• Knowledge of traditional machine learning and generative AI, hands-on experience with AI/ML frameworks, and experience with large-scale distributed data processing.

Nice to have

• Experience maintaining large-scale ML infrastructure and deployment tooling such as Ray, Kubernetes, or MLflow.
• Experience using generative AI and LLM assistants such as Claude or Gemini across the software development lifecycle.

Pay

$184,700 and $324,800 USD base salary; $175,000 and $308,500 USD base salary

Tools

Python, Java, Scala, PyTorch, TensorFlow, JAX, Ray, Spark

AI focus

Build AI/ML foundations and solutions for training, deploying, and monitoring models, with emphasis on security, privacy, performance, and scale.