(Summary generated by AI based on the full job description)
The project focuses on developing and optimizing ML kernels within a CoreML team aimed at improving training and serving performance and enabling the open-source ecosystem. Core technologies include C++, Python, TPU, GPU and low-level kernel languages like Pallas, Triton and optionally CUDA, plus frameworks JAX/PyTorch and compiler toolchains such as XLA/MLIR. Responsibilities cover designing and optimizing high-performance kernels, building benchmarking and autotuning infrastructure, performance analysis, regression testing, collaborating with researchers and compiler engineers, and enabling OSS adoption.

Optional


By clicking "Aplikuj" you confirm that you've read and accepted our Terms and Conditions.
This is how the employer processes your data
This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
Need more information?