Titan AI LogoTitan AI

mesh-transformer-jax

6,350
889
Python

Project Description

Model parallel transformers in JAX and Haiku

Project Information

Created on 3/13/2021
Updated on 9/21/2025