bhavnicksm / vanilla-transformer-jax

JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
โ˜†14Updated 3 years ago

Alternatives and similar repositories for vanilla-transformer-jax:

Users that are interested in vanilla-transformer-jax are comparing it to the libraries listed below