Machine Learning Platform and Infrastructure Engineer

4 days ago

Cupertino, California, United States Apple Inc. Full time
Job Summary

We are seeking a highly skilled Senior Machine Learning Engineer to join our On-Device Machine Learning team at Apple Inc. The successful candidate will be responsible for designing and implementing features that accelerate and compress large state-of-the-art models in our on-device inference stack.

Key Responsibilities
  • Design and implement features for our on-device inference stack to support accuracy-preserving techniques that empower model developers to compress and accelerate state-of-the-art models.
  • Build machine learning compilers, runtimes, execution kernels, and optimizations on ML models, as well as tooling for debugging and visualization of ML models.
  • Convert models from high-level ML frameworks to target devices (CPU, GPU, Neural Engine) for optimal functional accuracy and performance.
  • Write unit and system integration tests to ensure functional correctness and avoid performance regressions.
  • Diagnose performance bottlenecks and work with hardware and software architecture teams to co-design solutions that improve latency, power, and memory footprint of neural network workloads.
  • Analyze the impact of model optimization on model quality by partnering with modeling and adaptation teams across diverse product use cases.
Requirements
  • Bachelor's, Master's, or Ph.D. in Computer Science or related fields.
  • At least 4 years of experience in Machine Learning Engineering, System Software Engineering, or related fields.
  • Strong proficiency in C/C++ and Python.
  • Familiarity with ML fundamentals.
  • Familiarity with developing or using ML frameworks.
Preferred Qualifications
  • Experience with PyTorch/JAX Machine Learning frameworks.
  • Experience with machine learning model inference (serving) is a plus.
  • Experience with MLIR/LLVM compiler technologies is a plus.
  • Experience with on-device machine learning or system software design is a plus.
  • Experience with Swift is a plus.
  • Passion for designing software systems, APIs, and extensible products.
What We Offer
  • Comprehensive medical and dental coverage.
  • Retirement benefits.
  • A range of discounted products and free services.
  • Reimbursement for certain educational expenses.
  • Discretionary bonuses or commission payments.
  • Relocation assistance.
Apple is an Equal Opportunity Employer

We are committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.

  • AI Engineer, Machine Learning Platform

    4 days ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled AI Engineer to join our Foundation Model Infrastructure team within the Machine Learning Platform Technologies organization at Apple Inc.Key Responsibilities:Work alongside the Foundation Model Research team to optimize inference for cutting-edge model architectures.Collaborate with product teams to build...

  • AIML - Principal Software Engineer (Production Engineering), Machine Learning Platform and Infrastructure

    4 weeks ago

    Cupertino, California, United States Apple Full time

    AIML - Principal Software Engineer (Production Engineering), Machine Learning Platform and InfrastructureCupertino,California,United StatesMachine Learning and AIImagine what you could do with AIML at Apple. The people here don't just build products — we craft the kind of wonder that's revolutionized entire industries. It's the diversity of those people...

  • Machine Learning Infrastructure Engineer

    4 days ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Machine Learning Infrastructure Engineer to join our team at Apple Inc. As a key member of our team, you will be responsible for designing and building innovative and intelligent tools for our developers.Key ResponsibilitiesDevelop proficiency with and knowledge of the Swift programming language and...

  • Machine Learning Infrastructure Engineer

    5 days ago

    Cupertino, California, United States Apple, Inc. Full time

    About the RoleWe are seeking a highly skilled Machine Learning Infrastructure Engineer to join our team at Apple, Inc. As a key member of our team, you will be responsible for designing and building innovative and intelligent tools for our developers.Key ResponsibilitiesDevelop proficiency with and knowledge of the Swift programming language and...

  • Machine Learning Infrastructure Engineer

    13 hours ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Machine Learning Infrastructure Engineer to join our team at Apple Inc. As a key member of our software development team, you will play a critical role in designing and building innovative machine learning tools and frameworks that enable our developers to write, build, and test their code more efficiently.Key...

  • Senior Software Engineer

    7 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Software Engineer to join our Apple Maps Data Infrastructure team. As a key contributor, you will play a crucial role in shaping the technological landscape of our hybrid-cloud environment.Key ResponsibilitiesDesign and develop innovative solutions for machine learning infrastructure, batch processing,...

  • Machine Learning Engineer

    13 hours ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at Apple Inc. in Cupertino, California. As a Machine Learning Engineer, you will be responsible for designing, developing, and deploying machine learning models and systems that drive business value and innovation.Key ResponsibilitiesDesign and implement scalable machine...

  • Senior Software Engineer

    4 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Senior Software Engineer to join our dynamic team at Apple, where you will play a key role in enabling the next generation of intelligent experiences in our products and services.Key ResponsibilitiesDesign, build, and maintain large-scale distributed systems to support the end-to-end machine learning (ML)...

  • Software Development Engineer

    4 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Software Development Engineer to join our Apple Services Engineering team. As a key member of our iCloud Services SRE team, you will be responsible for designing and building large-scale distributed systems, particularly machine learning (ML) infrastructure and services.Key ResponsibilitiesSupport and maintain ML...

  • Senior Machine Learning Engineer

    13 hours ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our Contextual Understanding Team at Apple Inc. As a key member of this team, you will be responsible for developing and deploying cutting-edge machine learning models that power Siri and other on-device intelligence features.Key ResponsibilitiesDesign and implement...

  • Senior Machine Learning Engineer

    5 days ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Senior Machine Learning Engineer to join our Contextual Understanding Team at Apple Inc. As a key member of this team, you will be responsible for developing and deploying cutting-edge machine learning models that power Siri and other on-device intelligence features.Key ResponsibilitiesDesign and implement...

  • Machine Learning Resource Management Engineer

    4 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Machine Learning Resource Management Engineer to join our System Intelligent and Machine Learning (SIML) group at Apple. As a key member of our team, you will play a critical role in designing and leading infrastructure strategy and projects that guarantee a high level of service in a context of growing and...

  • Senior Software Data Infrastructure Engineer

    6 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled and experienced Senior Software Data Infrastructure Engineer to join our Data and Machine Learning Innovation organization at Apple. As a key member of our team, you will play a critical role in building and implementing a robust data processing framework to streamline log data pipelines.Key...

  • Machine Learning Platform and Technologies

    3 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Technical Program Manager to join our Machine Learning Platform and Technologies (MLPT) team in AI/ML. As a Technical Program Manager, you will play a critical role in simplifying and accelerating the adoption of machine learning in Apple products and ecosystems.Key ResponsibilitiesPartner with teams across Apple...

  • Data Infrastructure Engineer

    3 days ago

    Cupertino, California, United States Apple Full time

    About the RoleWe are seeking a highly skilled Data Infrastructure Engineer to join our AI and Machine Learning team at Apple. As a key member of our team, you will play a pivotal role in designing and implementing a robust data processing framework to streamline log data pipelines and build a flexible data insights infrastructure.Key ResponsibilitiesDesign...

  • Machine Learning Engineer- MSI

    1 month ago

    Cupertino, California, United States Apple, Inc. Full time

    Imagine what you could do here. At Apple, we believe new insights have a way of becoming excellent products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish. The people here at Apple don't just create products - they create the kind of wonder that's revolutionized...

  • Senior Software Engineer

    1 week ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking an experienced software engineer to join our Apple Data Platform team. As a Senior Software Engineer, you will play a key role in designing, implementing, and maintaining distributed systems to build world-class machine learning platforms and products at scale.Key ResponsibilitiesDesign and DevelopmentDesigning and implementing...

  • Machine Learning Compiler Engineer II

    5 days ago

    Cupertino, California, United States Amazon Web Services (AWS) Full time

    About the RoleWe are seeking a highly skilled Compiler Engineer II to join our Machine Learning Compiler team at Amazon Web Services (AWS). As a key member of this team, you will be responsible for designing and developing a deep learning compiler stack that takes neural network descriptions created in frameworks such as TensorFlow, PyTorch, and JAX, and...

  • Machine Learning Engineer, OS Intelligence

    22 hours ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking an exceptional Senior Applied Machine Learning Engineer to join our OS Intelligence team within Core OS Power & Performance at Apple Inc. As a key member of our team, you will design and develop innovative Machine Learning solutions that make Apple products more intelligent and personalized.Key ResponsibilitiesDesign and...

  • AI/ML - Machine Learning Engineer, Siri Understanding

    4 days ago

    Cupertino, California, United States Apple Inc. Full time

    About the RoleWe are seeking a highly skilled Machine Learning Engineer to join our team at Apple Inc. in the development of Siri, a revolutionary voice assistant that is redefining mobile computing.Key ResponsibilitiesDesign and develop advanced machine learning algorithms and models to improve the speech and audio experience of Siri.Collaborate with...