Show HN: How does JAX allocate memory on a TPU? An interactive C++ walkthrough
At the risk of sounding like a HN stereotype, for those who don&#x27;t know what a TPU is:<p>&gt; TPUs are hardware accelerators specialized in deep learning tasks. They are supported in Tensorflow 2.1 both through the Keras high-level API, and at a lower level.