Pushing the limits of GPU performance with XLA

Hi, I am the tech lead for this project, happy to answer questions.