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.2a0.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 83f8a7edac132e118c1d89e60eb3b54e9ad7a0deea07bf1418644a025c0831a9 |
|
| MD5 | 3eaed4ed693aba7599a4d1a4cebb5fe9 |
|
| BLAKE2b-256 | c72dacd2221a00d15625c7e778b74df030eaf01b418dfad16f9539f2934fb1fe |
Close
Hashes for vanilla_transformer_jax-0.0.2a0-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 4d1e3c5973a08c37bcc498e9013d7fb114a40d552842377d8480de52f7bd67ce |
|
| MD5 | 26e6ac6943e7f996d4843269db01b2c4 |
|
| BLAKE2b-256 | 7c4e5abfd30d5a57e07b80b096daefbe2adb68d79fbfa6d4db3e177b64463deb |
