9

Wrote a blog while learning distributed training with jax, wrote about the main confusions I got into (might as well be common!). Have a look: https://rish-av.github.io/blog/2025/jax_distributed/.