ML Platform Architect for Silicon & SoC

Apple, Inc.; Cupertino (CA); On-site; USD 216,000 - 325,000; Full time

Job summary

Apple is seeking a Senior ML Architect to map ML workloads to Apple silicon and develop performance models. You will collaborate across IP/SoC, system, software, and algorithm teams to optimize ML workloads on Neural Engine, GPU, and CPU.

You will evaluate model optimization techniques, design experiments, and create high-level complexity models to guide hardware direction. Strong C/C++/Python skills and experience with ML frameworks are required.

Qualifications

• Experience in C/C++ and/or Python.
• Experience in hardware IPs: ML HW accelerators, GPU/CPU, image/video processors or similar
• Experience with ML frameworks (e.g. PyTorch) and efficient implementations of machine learning algorithms

Responsibilities

• Create optimized implementations of ML workloads on Apple silicon including Neural Engine, GPU, and CPU.
• Collaborate with IP and SoC architecture teams to develop performance models and simulations of future hardware.
• Collaborate with system teams to create high-level performance models of emerging ML techniques and analyze system architecture trade-offs.
• Evaluate emerging ML model optimization techniques through experimentation and analysis; propose new ideas to inform hardware and algorithm direction.

Skills

C/C++

Python

ML HW accelerators

Education

MS or PhD in EE/CE/CS or related field

Tools

PyTorch

Job description

Apple is seeking a Senior ML Architect to map ML workloads to Apple silicon and develop performance models. You will collaborate across IP/SoC, system, software, and algorithm teams to optimize ML workloads on Neural Engine, GPU, and CPU.

You will evaluate model optimization techniques, design experiments, and create high-level complexity models to guide hardware direction. Strong C/C++/Python skills and experience with ML frameworks are required.