Machine Learning Engineer • Mid-Level

Who you are

You are a Machine Learning Engineer with a strong background in optimizing inference for large-scale models — you thrive on pushing the boundaries of technology and are eager to work on projects that impact billions of users. You have experience with frameworks like TensorFlow and PyTorch, and you are familiar with CUDA for performance optimization. You enjoy collaborating with research teams and product teams to bring innovative solutions to life.

You have a passion for building best-in-class technology and are not afraid to take risks — your boldness and ambition drive you to break the status quo and create impactful products. You understand the importance of low-latency services and are excited about the opportunity to work on Apple’s foundation models that power various services like Apple Search and Siri.

What you'll do

In this role, you will work alongside the Foundation Model Research team to optimize inference for cutting-edge model architectures — your contributions will directly impact the performance of models serving millions of customers in real time. You will build tools to identify bottlenecks in inference and collaborate closely with product teams to develop production-grade solutions. Your work will involve optimizing billions of parameters in language, vision, and speech models, ensuring they run efficiently at scale.

You will have the opportunity to innovate and implement state-of-the-art technologies that enhance the capabilities of Apple’s services — your role will be crucial in ensuring that the infrastructure you build can handle the demands of millions of queries every day. You will also engage in continuous learning and development, staying updated with the latest advancements in machine learning and AI.

What we offer

Apple offers a dynamic work environment where you can make a significant impact on technology that touches the lives of billions — you will be part of a team that values creativity and innovation. We encourage you to apply even if your experience doesn't match every requirement, as we believe diverse perspectives lead to better solutions. Join us at Apple and help shape the future of intelligent computing.