/transformers-jax

Built transformers from scratch in Jax

Primary LanguageJupyter Notebook

transformers-jax

Built transformers from scratch in Jax