Excited for this! Looking forward to finally getting hands-on with TPU architecture and seeing how JAX handles distributed training in practice.