Finding the best job has never been easier
Apple ML Compute Acceleration Engineer United States, California, Cupertino 402259090
06.05.2025
Share
- Responsibilities:- Adding optimizations in machine learning computation graph.- Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms.- 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.- Intended deliverables:- GPU Compute acceleration technology.- Optimized compute graphs across products.
- Proven programming and problem-solving skills.
- Good understanding of machine learning fundamentals.
- GPU compute programming models & optimization techniques.
- GPU compute framework development, maintenance, and optimization.
- Experience with system level programming and computer architecture.
- Experience with high performance parallel programming, GPU programming or LLVM/MLIR compiler infrastructure is a plus.
- Background in mathematics, including linear algebra and numerical methods.
- Strong communication and collaboration skills.
- Strong background of building high performance, production quality software on schedule.
- Experience with compiler technologies.
- Experience with adding computational graph support, runtime or device backend to Machine learning libraries (TensorFlow, PyTorch or JAX) support is a plus.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
