🥳 google/flaxformer: transformer architectures in JAX/Flax

[8bitmp3]

In case you haven't seen this...

https://github.com/google/flaxformer

Flaxformer: transformer architectures in JAX/Flax

Flaxformer is a transformer library for primarily NLP and multimodal research at Google. It is used for many NLP research use cases, providing both off-the-shelf BERT and T5 models, and several research projects built on shared components.