bhavnicksm / vanilla-transformer-jaxView on GitHub
JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
15Aug 16, 2021Updated 4 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. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?