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.1a0.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 593913342b3e2d4ee9fb2d8c8aadd413c505eb10d6f003aab871d16375c53f0f |
|
| MD5 | 2a5793851f0520965e588306b4e39525 |
|
| BLAKE2b-256 | ff5d53acb0c6d3ee95fff1e877f94156a8193eae66666564374e4531ab9130ee |
Close
Hashes for vanilla_transformer_jax-0.0.1a0-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 73362fea28ca3c9ceecdc84192b6d094474f5c1458969fc6aa7b59f99e7db1aa |
|
| MD5 | 9ed9a90b0dd43fc5479889db4f3b2b67 |
|
| BLAKE2b-256 | 7e6a3b2fa8730e73d9e8be26194db4520c08086d81b5dbc0184ccc5e37145bc6 |
