Memory Safe Computations with XLA Compiler

Publication
Advances in Neural Information Processing Systems (NeurIPS)