discover · compare · collections · requirements · contribute · feed · watching · stats · help with your tokens
kingoflolz/mesh-transformer-jax
Model parallel transformers in JAX and Haiku
History and source changes · Reports and owner responses · Compare