Memory safe computations with XLA compiler

Neural Information Processing Systems 

Software packages like TensorFlow and PyTorch are designed to support linear algebra operations, and their speed and usability determine their success.