ML Compute Acceleration Engineer
Apple
ML Compute Acceleration Engineer
Job Description
Apple’s Compute Frameworks team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV. Our efforts are currently focused in the key areas of linear algebra, image processing, machine learning, along with other projects of key interest to Apple. We are always looking for exceptionally dedicated individuals to grow our outstanding team to lay the foundation of technologies like Apple Intelligence. Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural Engine. Role has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine architectures.
Locations
- Cupertino, California, United States 95014
Salary
Estimated Salary Rangemedium confidence
30,000,000 - 60,000,000 INR / yearly
Source: ai estimated
* This is an estimated range based on market data and may vary based on experience and qualifications.
Skills Required
- machine learningintermediate
- GPU programmingintermediate
- linear algebraintermediate
- image processingintermediate
- data parallel algorithmsintermediate
Required Qualifications
- Proven programming and problem-solving skills. (experience)
- Good understanding of machine learning fundamentals. (experience)
- GPU compute programming models & optimization techniques. (experience)
- GPU compute framework development, maintenance, and optimization. (experience)
- Experience with system level programming and computer architecture. (experience)
- Experience with high performance parallel programming, GPU programming or LLVM/MLIR compiler infrastructure is a plus. (experience)
Preferred Qualifications
- Background in mathematics, including linear algebra and numerical methods. (experience)
- Strong communication and collaboration skills. (experience)
- Strong background of building high performance, production quality software on schedule. (experience)
- Experience with compiler technologies. (experience)
- Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support is a plus. (experience)
Responsibilities
- Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural Engine.
- Role has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine architectures.
- Adding optimizations in machine learning computation graph.
- Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms.
- Developing and maintaining MLIR<MetalML> dialect in Apple and open source with upgrades using latest LLVM.
- Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
- Tune GPU and Neural Engine accelerated compute across products.
- Tuning the cost model and optimizing runtime dispatch to multiple IPs to get best performance on Apple Silicon.
Tags & Categories
Answer 10 quick questions to check your fit for ML Compute Acceleration Engineer @ Apple.
10 Questions
~2 Minutes
Instant Score
Related Books and Jobs
ML Compute Acceleration Engineer
Apple
ML Compute Acceleration Engineer
Job Description
Apple’s Compute Frameworks team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV. Our efforts are currently focused in the key areas of linear algebra, image processing, machine learning, along with other projects of key interest to Apple. We are always looking for exceptionally dedicated individuals to grow our outstanding team to lay the foundation of technologies like Apple Intelligence. Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural Engine. Role has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine architectures.
Locations
- Cupertino, California, United States 95014
Salary
Estimated Salary Rangemedium confidence
30,000,000 - 60,000,000 INR / yearly
Source: ai estimated
* This is an estimated range based on market data and may vary based on experience and qualifications.
Skills Required
- machine learningintermediate
- GPU programmingintermediate
- linear algebraintermediate
- image processingintermediate
- data parallel algorithmsintermediate
Required Qualifications
- Proven programming and problem-solving skills. (experience)
- Good understanding of machine learning fundamentals. (experience)
- GPU compute programming models & optimization techniques. (experience)
- GPU compute framework development, maintenance, and optimization. (experience)
- Experience with system level programming and computer architecture. (experience)
- Experience with high performance parallel programming, GPU programming or LLVM/MLIR compiler infrastructure is a plus. (experience)
Preferred Qualifications
- Background in mathematics, including linear algebra and numerical methods. (experience)
- Strong communication and collaboration skills. (experience)
- Strong background of building high performance, production quality software on schedule. (experience)
- Experience with compiler technologies. (experience)
- Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support is a plus. (experience)
Responsibilities
- Our team is seeking extraordinary machine learning and GPU programming engineers who are passionate about providing robust compute solutions for accelerating machine learning networks on Apple Silicon using GPU and Neural Engine.
- Role has the opportunity to influence the design of compute and programming models in next generation GPU and Neural Engine architectures.
- Adding optimizations in machine learning computation graph.
- Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms.
- Developing and maintaining MLIR<MetalML> dialect in Apple and open source with upgrades using latest LLVM.
- Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
- Tune GPU and Neural Engine accelerated compute across products.
- Tuning the cost model and optimizing runtime dispatch to multiple IPs to get best performance on Apple Silicon.
Tags & Categories
Answer 10 quick questions to check your fit for ML Compute Acceleration Engineer @ Apple.
10 Questions
~2 Minutes
Instant Score
Related Books and Jobs