Probably the best source https://jax-ml.github.io/scaling-book/