maiush / jaxGPT Public

Implementation of GPT-2 in JAX

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.gitignore		.gitignore
README.md		README.md
encoder.py		encoder.py
gpt_jnp.py		gpt_jnp.py
gpt_np.py		gpt_np.py
utils.py		utils.py
workbook.ipynb		workbook.ipynb

Repository files navigation

jaxGPT

Implementation of GPT-2 in JAX

Implementation of GPT-2 in JAX

No releases published

No packages published