GPU ML Engineer

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

• Adding optimized GPU compute kernels across Machine Learning, Image Processing, Linear Algebra and Computer Vision
• Defining and implementing APIs in Metal Performance Shaders
• Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families
• Partnering with Platform Architecture teams to define Apple GPU's compute hardware roadmap
• Working with hardware team to analyze performance on future silicon
• Tune GPU-accelerated compute across products

• GPU compute kernel framework development, maintenance, and optimization
• Experience with system level programming and computer architecture
• Experience with high performance parallel programming, GPU programming experience

• Comprehensive medical and dental coverage
• Retirement benefits
• Range of discounted products and free services
• Reimbursement for certain educational expenses
• Discretionary bonuses or commission payments
• Relocation assistance

Apple’s Compute Frameworks team in Graphics, Games and Machine Learning org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV