nilutz / transformers_in_jax

Educational, straightforward implementation of vanilla, bert and GPT transformer models in jax

Geek Repo

Github PK Tool