Memory safe computations with XLA compiler. - Artem Artemev; Yuze An; Tilman Roeder; Mark van der Wilk (NeurIPS, 2022)

Memory Safe Computations with XLA Compiler. - Artem Artemev; Tilman Roeder; Mark van der Wilk (CoRR, 2022)