Apple - Cupertino, CA
posted 3 months ago
The Siri team in the AIML group at Apple is seeking an exceptional Machine Learning Engineer to lead efforts in identifying bottlenecks and optimizing our model inference stack. In this highly collaborative role, you will be at the center of multiple initiatives to accelerate and optimize large language models (LLMs) and other machine learning models used by Siri. This position involves consulting with multiple product teams to determine the appropriate foundation model (On Device vs Server) for their use cases and to help them achieve their accuracy and performance targets. Your work will directly impact Siri's performance and efficiency, enhancing the overall user experience. You will be expected to bring innovative ideas and a problem-solving mindset to tackle the unique challenges associated with optimizing complex ML models. As a Machine Learning Performance Engineer, you will play a critical role in ensuring the efficiency and scalability of Siri's machine learning models. You will work closely with diverse teams to diagnose performance issues and develop innovative solutions that enhance model performance. Your expertise will be pivotal in shaping the future of Siri's AI capabilities. You will analyze and optimize the performance of machine learning models and systems used by Siri, develop and implement strategies for model tuning, parameter optimization, and efficient resource usage. Additionally, you will conduct performance benchmarking and develop tooling and metrics to measure model performance in terms of compute, memory, and latency. Collaboration with feature and product teams will be essential to consult on modeling decisions to achieve Siri performance objectives, as well as working with hardware and software teams to integrate research findings into product implementation.