Staff Machine Learning Engineer

Apple

United States Of America, United States
Not specified (assumed hybrid or onsite based on company culture).
End-to-end machine learning workflow expertise
Large scale distributed systems experience
Generative techniques like transformers and diffusion
Apple is seeking a Staff Machine Learning Engineer to join their team focused on ML infrastructure and generative AI. The role involves developing robust systems that enhance machine learning workflows, from data preparation to deployment, within a collaborative and innovative environment

Job Summary

  • Join a team at the forefront of ML infrastructure and generative AI to enable next-generation intelligent experiences on Apple products.
  • You will build robust systems connecting scalable data pipelines with advanced ML workflows to accelerate real-world AI applications.
  • The role offers autonomy to experiment with generative models while working alongside a world-class team to ship high-quality AI solutions.

Matching Summary

Match Score: 85

Apple is seeking a Staff Machine Learning Engineer to join their team focused on ML infrastructure and generative AI. The role involves developing robust systems that enhance machine learning workflows, from data preparation to deployment, within a collaborative and innovative environment.

Skills & Requirements

Must-have

  • End-to-end machine learning workflow expertise
  • Large scale distributed systems experience
  • Generative techniques like transformers and diffusion
  • Production environment data and ML infrastructure
  • Java, Python, or Go programming proficiency

Nice-to-have

  • PyTorch, JAX, or TensorFlow framework proficiency
  • Columnar and lakehouse format knowledge
  • Performance engineering for I/O-bound workloads
  • Data lineage and governance system experience
  • Spark, Daft, Polars, or DuckDB internals

Key Requirements

  • B.S., M.S., or Ph.D. in Computer Science or equivalent
  • Strong foundation in end-to-end ML lifecycle
  • Experience designing highly available scalable systems

Work Rights

Not specified

Tailored Resume

Cover Letter