@inproceedings{NEURIPS2022_782b6152, author = {Artemev, Artem and An, Yuze and Roeder, Tilman and van der Wilk, Mark}, booktitle = {Advances in Neural Information Processing Systems}, editor = {S. Koyejo and S. Mohamed and A. Agarwal and D. Belgrave and K. Cho and A. Oh}, pages = {18970--18982}, publisher = {Curran Associates, Inc.}, title = {Memory safe computations with XLA compiler}, url = {https://proceedings.neurips.cc/paper_files/paper/2022/file/782b6152c04e9948c2cb3833e9a288ef-Paper-Conference.pdf}, volume = {35}, year = {2022} }