2 ms·

https://colab.research.google.com/github/kingoflolz/mesh-transformer-jax/blob/master/colab_demo.ipynb https://colab.research.google.com/github/kingoflolz/mesh-t

https://colab.research.google.com/github/kingoflolz/mesh-transformer-jax/blob/master/colab_demo.ipynb https://colab.research.google.com/github/kingoflolz/mesh-tra...