GitHub repository: mplemay/flax-transformer
Primary language: Python
Created: 2021-07-07T21:30:33Z
Description: A flax implementation of the transformer architecture proposed in "Attention Is All You Need"