Show HN: How does JAX allocate memory on a TPU? An interactive C++ walkthroughgist.github.com78 pointssillysaurusx5 years ago