This role pays more than
92%
of similar roles. Most pay
$162,000–$241,750
— the shaded band above.
At the midpoint, this role pays about
$255k
versus about
$202k
for comparable roles.
Based on 240 similar postings.
Employer
About Apple Inc
Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software
Apple Inc currently has
3587 open roles
on FindRole.
Listed pay typically runs
$165,800–$277,600
across 2768 roles with salary data.
SW Optimization Engineer AI/ML joins the Platform Architecture group to integrate hardware and software into a unified system. This role focuses on identifying performance bottlenecks in new and existing workloads while providing recommendations for hardware and software improvements. You will conduct performance studies to validate architecture decisions, create optimized implementations of machine learning workloads on Apple Silicon—including the Neural Engine, GPU, and CPU—and develop performance models for emerging AI/ML techniques. Collaborating with system and tool teams, you will deliver analysis instruments and optimized libraries for AI/ML applications. The role requires expertise in C/C++, Python, and PyTorch, alongside knowledge of computer architecture, operating system internals, and compiler technologies. You will solve complex problems regarding performance trade-offs and develop high-level complexity models or low-level performance libraries for specialized hardware accelerators.
Identify performance bottlenecks in hardware and software across existing and new workloads.
Create optimized implementations of machine learning workloads on CPU, GPU, and Neural Engine.
Conduct performance studies to inform and validate architectural decisions for Apple Silicon.
Develop performance models for emerging AI/ML techniques and analyze system architecture trade-offs.
Deliver performance analysis instruments, optimized libraries, and frameworks for AI/ML applications.
Provide feedback to silicon and software teams based on identified patterns of performance challenges.
Prototype and benchmark algorithms while analyzing performance metrics and creating high-level complexity models.
Develop highly efficient low-level performance libraries for AI/ML accelerators and GPUs.
What we're looking for
Bachelor's degree or equivalent experience in Computer Engineering, Computer Science, or a related field.
Experience in software development for AI/ML hardware accelerators, GPUs, image/video encoders, or similar.
Experience with AI/ML, graphics, or HPC performance benchmarks and workloads.
M.S. or Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, or a closely related field (preferred).
10+ years of experience in software performance optimization, analysis tools, processes, and developing efficient computational algorithms (preferred).
Knowledge of computer architecture fundamentals (preferred).
Proficiency in C/C++ family programming languages and scripting languages like Python (preferred).
Proficiency in AI/ML frameworks like PyTorch and knowledge of OS internals or compiler technologies (preferred).
System Power Modeling
AI Systems Architecture
C++
Python
Perl
Ruby
Scripting
Platform Management Framework
Power Analysis
Thermal Constraints
Oscilloscopes
Logic Analyzers
Validation
Automation
Hardware Modeling
Artificial Intelligence
Machine Learning
Python
Big Data
Statistical Techniques
Data Transformation
Model Selection
Cross-Validation
System Architecture
Systems Modeling
Simulation
Data Engineering