vanilla-transformer-jax (pypi): JAX/Flax implimentation of 'Attention is All You Need' by Vaswani et al.
Keywords: transformer|JAX|Flax|Deep-Learning|artificial-intelligence|natural-language-processing