Machine Learning Engineer • Senior

Who you are

You have 5+ years of experience in machine learning and platform engineering, with a strong focus on building scalable systems that support diverse ML frameworks. Your expertise in Python allows you to design SDKs and APIs that enhance the developer experience, making it easier for ML practitioners to navigate datasets and model serving. You thrive in collaborative environments, working closely with ML engineers to understand their workflows and identify pain points. Your problem-solving skills enable you to tackle complex technical challenges that span multiple systems, ensuring seamless integrations across the ML platform.

You are well-versed in various ML frameworks, including PyTorch and JAX, and understand the nuances of integrating these tools into a cohesive platform. Your experience in backend services and model management infrastructure equips you to build robust solutions that empower teams to focus on innovation rather than infrastructure complexity. You are passionate about enhancing developer productivity and creating a unified experience across different ML tools.

Desirable

Experience with additional ML frameworks or tools is a plus, as is familiarity with cloud-based infrastructure and services. You are open to learning new technologies and adapting to the evolving landscape of machine learning and AI.

What you'll do

In this role, you will be responsible for building and scaling core platform capabilities that support model management and serving infrastructure. You will create seamless integrations with the ML frameworks that Apple's teams depend on, ensuring that the platform feels native to each framework while providing a unified experience. Your work will involve designing and implementing backend services, developing Python SDKs, and creating APIs that facilitate smooth interactions between ML practitioners and the platform.

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 delivering innovative AI-driven experiences across Apple's hardware and software products. Your contributions will directly impact the development of Apple Intelligence and other intelligent experiences, making your role crucial to the success of the organization.

What we offer

Apple provides a dynamic work environment where you can contribute to cutting-edge technology and work alongside talented professionals in the field. You will have the opportunity to grow your skills and advance your career while making a significant impact on the future of machine learning at Apple. The company values diversity and inclusion, fostering a culture where all employees can thrive and contribute their unique perspectives. Competitive compensation and benefits are part of the package, ensuring that you are rewarded for your contributions to the team.