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.1a1.tar.gz
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | d68dc84d62ec73d4744a86db54829a98d328cb5bb5a6937be1ac3bba0bfd1612 |
|
| MD5 | 7d8d59dff10593660da1db23591183e0 |
|
| BLAKE2b-256 | eac42404d8266653d8cf97576f11434bec842efc2663fe0172dcecc2d04f90c5 |
Close
Hashes for vanilla_transformer_jax-0.0.1a1-py3-none-any.whl
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 0c3833b7f313c7cfb5d708603c92326e1d205f3aa72a401afe27b35ab3e55872 |
|
| MD5 | 4435a434b42ecb4461a6b6bbd1176673 |
|
| BLAKE2b-256 | c40945258450ecc3e43f3651e11d3089e4aac7c801640adc676e42cf5b6e658c |
