SW Optimization Engineer AI/ML
Core
Design and optimize machine learning workloads on Apple Silicon (Neural Engine, GPU, CPU) to address performance bottlenecks and inform system architecture decisions.
Role type
Senior IC SW Optimization Engineer (AI/ML)
Builds
Optimized implementations of ML workloads, performance analysis instruments, and efficient computational libraries for Apple Silicon.
Domain
Consumer electronics / AI/ML hardware acceleration
Deliverable
production ML models
Required skills
C/C++, Python, computer architecture fundamentals, software development for AI/ML HW accelerators/GPUs, performance modeling, algorithm prototyping and benchmarking
Preferred skills
PyTorch, OS internals, compiler technologies, HPC performance benchmarks, low-level library development for AI/ML accelerators
Technologies
Apple Silicon, Neural Engine, GPU, CPU, PyTorch
Responsibilities
Conduct performance studies to validate architecture decisions; Create optimized implementations of ML workloads; Collaborate with system teams to create performance models; Work with tools teams to deliver performance analysis instruments and optimized libraries.
Seniority
Senior, hands-on IC
