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.2.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 51d45b828cff06602ad5f4e3194c4667d15feb53659adf2e5db22831fd21e21b |
|
| MD5 | aa2f3714d17b12823fa99b84db963aa7 |
|
| BLAKE2b-256 | 2591ce5f712f7d8bae0fce51f5b16ba5e0e078f270f7e9741b77ae37739abc1b |
Close
Hashes for vanilla_transformer_jax-0.0.2-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | aa9fdb4bcb34b8ca74ef3f7f63535053cec3c4e7e0ea132bf61ddca7ac854e0e |
|
| MD5 | 5d9eccc5a5543cd0e9ac2d0146ca8018 |
|
| BLAKE2b-256 | 0b77917589c80dd9e82d8adbf2dd21723bb913519f5f34960b6ec4bcadc36a0c |
