Expoint – all jobs in one place
המקום בו המומחים והחברות הטובות ביותר נפגשים
Limitless High-tech career opportunities - Expoint

Apple Machine Learning Engineer Platform Architecture 
United States, West Virginia 
721643050

Yesterday
In this role, you will explore different ways of mapping ML workloads to Apple silicon and develop performance models/simulations. Your work will inform and validate architecture decisions. You will gain insights on how to make workloads run efficiently on our IPs and SoCs and communicate what we learn to software and algorithm teams.
• 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.• Conduct performance studies to inform and validate architecture decisions.• Collaborate with system team to create high level performance models of emerging ML techniques and analyze system architecture trade-offs.
  • Bachelor's degree
  • Ability to program in C/C++ and/or Python
  • Knowledge of computer architecture fundamentals
  • Domain knowledge in at least one hardware IP: ML HW accelerators or processing units such as GPU, image/video, CPUs, or similar
  • MS or PhD in EE/CE/CS or related field, or 3+ years of relevant experience
  • Experience in efficient implementation of machine learning algorithms
  • Experience in creating system or IP performance models/simulations
  • Verbal and written communication skills for collaborating with partner teams
  • Familiarity with deep learning frameworks such as PyTorch
  • Ability to prototype and benchmark algorithms on CPU/GPU/Neural Engine, analyze performance metrics and create high level complexity models
  • Ability to develop hardware accelerator performance and bit accurate models
  • Understanding of compiler frameworks/technologies
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.