Attention Is All You Need Implementing "Attention Is All You Need" Vaswani et al. from scratch in Jax and Torch.