jaxGPT

A Jax re-implementation of GPT with training and inference.

Name		Name	Last commit message	Last commit date
Latest commit History 87 Commits
examples		examples
exp		exp
jaxgpt		jaxgpt
.gitignore		.gitignore
README.md		README.md
requirements.txt		requirements.txt
setup.py		setup.py
test.py		test.py

Provide feedback