The simplest, fastest repository for training/finetuning medium-sized GPTs in Jax. - View it on GitHub