Show HN: How does JAX allocate memory on a TPU? An interactive C++ walkthrough | Heykuki News