AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

2 days ago


Cupertino, United States Apple Inc. Full time
AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure

The AIML - On-Device Machine Learning group is responsible for the creation of amazing on-device ML experiences. The team builds foundational machine learning frameworks and tools to optimize large language/vision/multi-modal models that power on-device ML features across Apple products and services. The group is looking for a senior software engineer to help define and implement features that accelerate and compress large state of the art (SoTA) models (e.g., LLMs) in our on-device inference stack. This is a unique opportunity to work on exciting new technologies and contribute to Apple's ecosystem, with a commitment to privacy and user experience impacting millions of users worldwide.Are you someone who can write high-quality, well-tested code and collaborate cross-functionally with partner Hardware, Software, Machine Learning, and Research teams across the company? Do you have any experience building Machine Learning compilers/runtimes/kernels/optimization tools? If so, come join us and be a part of the team that is helping Machine Learning developers innovate and ship enriching experiences on Apple devices

Description

This role sits at the intersection of software engineering and ML engineering. As a member of this team, the successful candidate will:- Build features for our on-device inference stack to support the most relevant accuracy preserving, general purpose techniques that empower model developers to compress and accelerate SoTA models (e.g., LLMs) in apps- This includes building Machine Learning compilers, runtimes, execution kernels, optimizations on ML models, tooling for debugging/visualization of ML models, etc.- Convert models from a high-level ML framework to a target device (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 HW & SW Arch teams to co-design solutions that further improve latency, power, and memory footprint of neural network workloads- Analyze impact of model optimization (compression/quantization etc) on model quality by partnering with modeling and adaptation teams across diverse product use cases.

Minimum Qualifications

  • Bachelors/Masters/PhD in Computer Science or related fields.
  • At least 4 years of experience in Machine Learning (ML) 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 big plus.
  • Experience with MLIR / LLVM compiler technologies is a big plus.
  • Experience with on-device machine learning or system software design is a big plus.
  • Experience with Swift is a plus.
  • Passion for designing Software systems, APIs, and extensible products.
  • At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $143,100 and $264,200, and your base pay will depend on your skills, qualifications, experience, and location.

    Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

    Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is 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. Learn more about your EEO rights as an applicant.

#J-18808-Ljbffr

  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Machine Learning Platform & InfrastructureCupertino, California, United StatesMachine Learning and AIImagine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there’s no telling what you could...


  • Cupertino, United States Apple Inc. Full time

    AIML - Machine Learning Engineer, Machine Learning Platform & Infrastructure magine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there’s no telling what you could accomplish.Do you want to make Siri and Apple products...


  • Cupertino, United States Apple Inc. Full time

    AIML-Sr. Machine Learning Engineer, MeasurementWe are seeking a highly skilled and experienced software engineer to join our team at Apple AIML in building an on-device machine learning and observability platform. As a key member of the team, you will help design and develop the infrastructure and work on various machine learning solutions and models. You...


  • Cupertino, United States Apple Full time

    SummaryWe are seeking a highly skilled and experienced software engineer to join our team at Apple AIML in building an on-device machine learning and observability platform. As a key member of the team, you will help design and develop the infrastructure and work on various machine learning solutions and models. You will also have the chance to collaborate...


  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Foundation Models Cupertino, California, United States Machine Learning and AI We are a group of engineers and researchers responsible for building foundation models at Apple. We build infrastructure, datasets, and models with fundamental general capabilities such as understanding and generation of text, images, speech,...


  • Cupertino, United States Apple Inc. Full time

    AIML - ML Engineer, Machine Learning Platform & Infra Machine Learning and AI To view your favorites, sign in with your Apple ID. Do you feel you think differently, you are eager to break status quo, are bold and ambitious, aren’t afraid to take risks and are passionate to build the best of class technology. If yes, what better place to be at and do...


  • Cupertino, United States Apple Full time

    AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & Technology Cupertino, California, United States Machine Learning and AI Do you want to shape the platform that enables the next generation of intelligent experiences on Apple products & services? In Apple’s Machine Learning Platform Technology & Infra team we have built the platform that...


  • Cupertino, United States Apple Inc. Full time

    AIML - Staff Machine Learning Engineer - ML Efficiency, ML Platform & TechnologyDo you want to shape the platform that enables the next generation of intelligent experiences on Apple products & services? In Apple’s Machine Learning Platform Technology & Infra team we have built the platform that Apple uses for developing machine learning, artificial...


  • Cupertino, United States Apple Inc. Full time

    AIML - Machine Learning Engineer, Foundation Models We are a group of engineers and researchers responsible for building foundation models at Apple. We build infrastructure, datasets, and models with fundamental general capabilities such as understanding and generation of text, images, speech, videos, and other modalities and apply these models to Apple...


  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Siri Information Intelligence Cupertino, California, United States Machine Learning and AI The AIML Information Intelligence team is creating groundbreaking technology for artificial intelligence, machine learning, and natural language processing! The features we create are redefining how hundreds of millions of people use...


  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Foundation Model ServicesCupertino, California, United StatesMachine Learning and AIDo you feel you think differently, you are eager to break status quo, are bold and ambitious, aren’t afraid to take risks and are passionate to build the best of class technology? If yes, what better place to be at and do this than Apple?...


  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Siri Conversation IntelligenceCupertino, California, United StatesMachine Learning and AIThe Siri Conversation Intelligence (SCI) team is creating groundbreaking technology for large scale systems, spoken language, machine learning, and LLMs to enable better conversational experience with Siri. The features we create are...


  • Cupertino, California, United States CoolSnail Full time

    Job DescriptionJob Title: Machine Learning Engineer/AI EngineerWe are seeking a highly skilled Machine Learning Engineer/AI Engineer to join our team at CoolSnail. As a key member of our team, you will be responsible for designing, developing, testing, deploying, maintaining, and enhancing Machine Learning Pipelines using K8s/AKS based Argo Workflow...


  • Cupertino, United States Apple Inc. Full time

    AIML - Staff Machine Learning Engineer, Retrieval and Ranking Quality (SII)We are looking for a Machine Learning engineering technical leader with a passion for using machine learning to create intelligent search applications. Our team researches and implements novel retrieval and ranking techniques, machine learning algorithms and models that power amazing...


  • Cupertino, United States Apple Full time

    AIML - Sr Machine Learning Engineer, Data and ML Innovation Cupertino, California, United States Machine Learning and AI Do you want to play a part in the next revolution in large language models, contribute to products that are redefining mobile and desktop computing, and work with the people who built the intelligent assistant and search products that help...


  • Cupertino, United States Apple Full time

    Machine Learning Engineer Cupertino, California, United States Software and Services We’re looking for industry-leading machine learning technologists to help shape the future of AI-driven system experiences across the Apple ecosystem. You’ll be joining a small, dynamic team that will develop novel ML techniques and applications, build tools and...


  • Cupertino, United States Apple Full time

    SummaryDo you feel you think differently, you are eager to break the status quo, are bold and ambitious, aren’t afraid to take risks and are passionate to build the best of class technology? If yes, what better place to be at and do this than Apple? At Apple, "we think different, we push the boundaries of computing and intelligence. We build products that...


  • Cupertino, United States Apple Inc. Full time

    Imagine what you could do here! The people here at Apple don’t just create products — they build the kind of wonder that’s revolutionized entire industries. It’s the diversity of those people and their ideas that inspires the innovation that runs through everything we do, from amazing technology to industry-leading environmental efforts. At Apple,...


  • Cupertino, United States Apple Full time

    AIML - Machine Learning Engineer, Siri and Information Intelligence Cupertino, California, United States Machine Learning and AI The Siri Attention and Invocation Data team is in search of a curious and innovative team player who is passionate about building and innovating our data strategy. Our team is responsible for all aspects of generating datasets to...


  • Cupertino, United States Apple Inc. Full time

    AIML - Sr Machine Learning Engineer, Data and ML Innovation As part of Apple's AI and Machine Learning org, we encourage and create groundbreaking technology for large-scale ML systems, computer vision, natural language processing, and multi-modal understanding. The Data and Machine Learning Innovation (DMLI) team is looking for a passionate Machine...