SW ML Optimization Engineer
Core
Analyze AI/ML workloads on Apple Silicon to identify performance bottlenecks and develop optimized software implementations.
Role type
ML optimization engineer (system software)
Builds
Optimized software workloads and performance analysis instruments for Apple Silicon
Domain
Consumer electronics / AI/ML / System-on-a-chip
Deliverable
production ML models
Required skills
GPU/parallel programming, profiling tools, Python/C/C++, instruction set optimization
Preferred skills
compute kernel tuning, ML frameworks, hardware/software co-design, low-level libraries
Technologies
Metal, OpenCL, CUDA, Xcode Instruments, VTune, Nsight Compute, PyTorch, MLX, Core ML
Responsibilities
Profile and analyze AI/ML workloads across Apple Silicon compute engines; Build targeted microbenchmarks to characterize device strengths and weaknesses; Develop and tune optimized implementations of software workloads; Partner with system teams to prototype performance experiments and build analysis libraries.
Seniority
Mid-level, hands-on IC
