Machine Learning Engineer • Senior

Who you are

You have 5+ years of experience in machine learning and platform engineering, with a strong background in building scalable systems that support diverse ML frameworks. Your expertise in Python allows you to design SDKs and APIs that enhance the productivity of ML practitioners. You thrive at the intersection of engineering and user experience, understanding the complexities of integrating various ML tools and frameworks seamlessly.

You possess deep knowledge of machine learning frameworks such as PyTorch and JAX, and you are comfortable collaborating with ML engineers to identify pain points and develop solutions that streamline their workflows. Your problem-solving skills enable you to tackle complex technical challenges that span multiple systems, ensuring that the ML platform remains robust and efficient.

What you'll do

In this role, you will be responsible for building and scaling core platform capabilities for model management and serving infrastructure. You will create seamless integrations with the ML frameworks that Apple's teams depend on, ensuring a unified experience across the platform. Your work will involve designing and implementing backend services, developing Python SDKs, and creating integrations that facilitate smooth transitions between datasets, training, model serving, and evaluation.

You will collaborate closely with ML engineers to understand their workflows and identify areas for improvement. By building platform features that enhance productivity, you will enable teams to focus on innovation rather than infrastructure complexity. Your contributions will directly impact the development of Apple Intelligence and other AI-driven experiences across iOS, macOS, and other platforms.

What we offer

Apple offers a dynamic work environment where innovation is at the forefront. You will have the opportunity to work with cutting-edge technologies and collaborate with talented professionals in the field. We encourage you to apply even if your experience doesn't match every requirement, as we value diverse perspectives and backgrounds. Join us in shaping the future of intelligent experiences at Apple.