Flaxformer is a transformer library for primarily NLP and multimodal research at Google.

Flaxformer: transformer architectures in JAX/Flax Flaxformer is a transformer library for primarily NLP and multimodal research at Google. It is used