(Summary generated by AI based on the full job description)
The project focuses on developing and optimizing ML kernels within the CoreML team for training and serving models. Key technologies include TPU, GPU, Pallas, Triton, CUDA, C++ and Python, with ML frameworks like JAX, PyTorch and compiler tools such as XLA/MLIR. Main responsibilities include designing and optimizing high-performance kernels, building benchmarking and autotuning infrastructure, performance analysis, regression testing and collaboration with ML researchers, framework developers and compiler engineers.

Optional


By clicking "Aplikuj" you confirm that you've read and accepted our Terms and Conditions.
This is how the employer processes your data
Certain terms used in this advertisement result from the standard layout and functionality of the website. They are technical in n ...
This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
Need more information?