Ml Infrastructure Engineer • Senior

Who you are

You have a strong background in machine learning infrastructure with a focus on generative AI technologies — your experience includes building and optimizing systems for training large-scale models that impact millions of users. You are technically hands-on and possess deep subject matter expertise in ML infrastructure, allowing you to effectively collaborate with algorithm design and infrastructure engineers.

You are skilled in training optimizations and profiling, particularly for vision and language pre-training — your knowledge extends to researching effective scheduling for multimodal training workloads, ensuring that the infrastructure can handle complex tasks efficiently. You thrive in a collaborative environment, coordinating with diverse teams to drive innovation and improve system performance.

What you'll do

In this role, you will be responsible for optimizing training algorithms and profiling for multimodal machine learning workloads — your work will involve experimenting with post-training ablations, including reward modeling and prompt optimization. You will benchmark and prototype algorithmic choices to ensure they align with the training and deployment infrastructure, contributing to the seamless experiences Apple users enjoy.

You will also engage in research to develop training recipes that enhance the efficiency of multimodal training workloads — your insights will help steer the direction of algorithmic choices, ensuring that the infrastructure is robust and scalable. As part of a multidisciplinary team, you will play a key role in shaping the future of generative AI at Apple, impacting how users interact with technology.

What we offer

Apple offers a dynamic work environment where innovation thrives — you will be part of a team that is at the forefront of generative AI technology, working on projects that enrich the lives of billions. We provide competitive compensation and benefits, along with opportunities for professional growth and development. Join us in creating groundbreaking technologies that redefine user experiences.