GitHub repository: jxnscott/attention-is-all-you-need-jax
Description: replication of the original transformer in JAX