Thanks for the suggestion! That's only for TPUs, or would it also work on Nvidia GPUs?

→ View original post on X — @rasbt