Machine Learning Engineer, Platform Architecture
Core
Designing system-on-a-chip architecture and mapping machine learning workloads to Apple silicon to create optimized, integrated solutions.
Role type
Senior IC machine learning engineer (platform architecture)
Builds
System-on-a-chip architectures and optimized ML implementations for Apple devices
Domain
Consumer electronics / Hardware architecture
Deliverable
production ML models
Required skills
C/C++ programming, Python programming, computer architecture fundamentals, ML hardware accelerator knowledge, performance modeling, simulation, system architecture trade-off analysis
Preferred skills
PyTorch, ML model deployment, runtime framework optimization, compiler frameworks, algorithm prototyping on CPU/GPU/Neural Engine, high-level complexity modeling
Technologies
Apple silicon, Neural Engine, GPU, CPU, PyTorch
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. Conduct performance studies to inform and validate architecture decisions. Collaborate with system teams to create high-level performance models of emerging ML techniques and analyze system architecture trade-offs.
Seniority
Senior, hands-on IC