https://hackershare.dev/en/bookmarks/763958
Show HN: How does Jax allocate memory on a TPU? An interactive C++ walkthrough