-Responsibilities: * Adding optimizations in machine learning computation graph. * Defining and implementing APIs in Metal Performance Shaders Graph, investigating new algorithms. * Develop/maintain/optimize ML training acceleration technologies. * Support adoption of GPU-accelerated training across 1st and 3rd party clients. * Tune GPU-accelerated training across products. * Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.- Intended deliverables: * ML training GPU acceleration technology. * Optimized ML training across products.