J jax-ml.github.io How to scale your model: A systems view of LLMs on TPUs 185 points mattjjatgoogle 1 year ago 30 comments
