JAX/Flax implimentation of 'Attention is All You Need' by Vaswani et al.
Project description
vanilla-transformer-jax
JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for vanilla-transformer-jax-0.0.3a0.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 47a7373f165200a85cbf62656a9da3ba2c1d58c47fa2a1c8f393f08724f2fa37 |
|
| MD5 | eb402ae90aff90e35f69ed3ad8a59536 |
|
| BLAKE2b-256 | 2e99cc9e2e2012581a2e091c3f178dd5322015a0c970520fab641ac3be8ff723 |
Close
Hashes for vanilla_transformer_jax-0.0.3a0-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 08e44f107f68180aecf2d0cd8d3ffaa43a6f1169dfaac7035b39311736efc73a |
|
| MD5 | e35f71f1f1278c1ff51dbdc3affb609c |
|
| BLAKE2b-256 | 661ac383895b7bfb1bae70d26f8266dfde7670e75bc41ad21c8c32264163d0bb |
