AIML - Senior Machine Learning Engineer - ML Hub Core, Machine Learning Platform Technologies (MLPT)

Imagine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Join us in enabling the next generation of intelligent customer experiences in Apple's products and services with the latest advancements in Generative AI and Large Language Models! We are seeking an experienced ML Engineer to join our dynamic team. You'll play a key role in designing, developing, and maintaining the central platform for building AI experiences within Apple. The ideal candidate has extensive experience building tools for ML lifecycle and launching ML-powered experiences to production. This candidate previously worked with ML frameworks, TPU/GPU/CPU workflows, and is knowledgeable of the Python ML ecosystem.

Description

We design and build machine learning infrastructures to support features that empowers billions of Apple users. As part of this group, you will develop the systems we use to build AI/ML models through the entire lifecycle - from experimentation to production inference. You will have opportunities to work cross functional with MLE, DS, Infra and others. You will be able to stay on the cutting edge of tooling that supercharges ML/AI innovation at Apple.

Preferred Qualifications

Experience working with notebook ecosystems and web-based IDEs. Experience with building and deploy ML software in production to a wide range of infrastructure (CPU/GPU/TPU).

Big data experience such as Apache Spark, Apache Ray

Some frontend development experience with Javascript / React

Prior experience building an Notebook solution in a large organization

Minimum Qualifications

Bachelors in Computer Science, engineering, or a related field

Proficiency in Python and advanced knowledge of the Python ML ecosystem

Extensive experience building tools for ML lifecycle and launching ML-powered experiences to production

Proven record of building responsive and scalable services

Experience with service oriented architecture, observability, database and distributed systems

Experience with building on top of AWS or GCP cloud services

Solid understanding of software engineering principles and design patterns

Strong communication skills and ability to work in a collaborative environment