AIML - Machine Learning Manager, ML Hardware/Software Co-Design, Machine Learning Platform & Infrastructure

Key Skills

Aartificial intelligence, C Programming, C++, Design, Effective communication skills, Infrastructure as code, Machine learning techniques, Python Programming, Streaming APIs

Job Description

The on-device Machine Learning Performance team in the Machine Learning Platforms and Infrastructure group is looking for an ML Manager to lead a team of engineers engaging in various aspects of hardware and software co-design. Team responsibilities include analyzing ML performance on unreleased SOCs (System On a Chip), signing off on new products’ ML performance, and investigating performance and power characteristics of new ML workloads. In addition, the team researches and develops toolchain to identify performance bottlenecks as well as experimenting with various techniques for model and runtime level techniques to meet theoretical targets. Additionally, the team owns cost modeling efforts with an intent to deliver the best performance across all compute backends. The team is highly cross functional and its leader will need to leverage its technical strengths to influence hardware and software engineering partners to help drive and shape Apple’s product roadmap. Experience with system level performance in embedded platforms as well as SOC architecture is a strong plus. If this role sounds exciting, we want to hear from you!

Key Qualifications

• Proficiency in C/C++ and Python
• Understanding of machine learning fundamentals
• Experience with performance on low level software and computer architecture.
• Passion for software architecture, APIs and high performance extensible software.
• Creative, collaborative, and product-focused
• Excellent communication skills

You will be tasked to analyze and signoff on ML performance and power on unreleased SOCs, product level analysis and signoff, and profiling of new ML workloads such as generative models. You will be responsible for collecting the takeaways from SOC investigations and then collaborating with the Platform Architecture team to provide insight for future silicon. You’ll work with high level software features including 3rd party developers, as well as teams up and down the ML inference stack all the way down to firmware. You’ll develop innovative methodology for measuring and introspecting into ML workloads and extend our tools and infra to provide deeper insight into system behaviors. You’ll also work with model developers to optimize their models and optimize their usage of our SOCs to deliver outstanding latency and memory usage. You'll manage a team a team building cost models used for model segmentation.

Education & Experience

B.S or M.S. Degree in Computer Science/Engineering, or equivalent work experience.

Additional Requirements

• Apple’s most important resource, our soul, is our people. Apple benefits help further the well-being of our employees and their families in meaningful ways. No matter where you work at Apple, you can take advantage of our health and wellness resources and time-away programs. We’re proud to provide stock grants to employees at all levels of the company, and we also give employees the option to buy Apple stock at a discount — both offer everyone at Apple the chance to share in the company’s success. You’ll discover many more benefits of working at Apple, such as programs that match your charitable contributions, reimburse you for continuing your education and give you special employee pricing on Apple products.
• Apple benefits programs vary by country and are subject to eligibility requirements.
• 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. Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities. Apple is a drug-free workplace.

Pay & Benefits

• 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 $183,400 and $316,900, and your base pay will depend on your skills, qualifications, experience, and location.