samyakai's repositories
mesh-transformer-jax
Model parallel transformers in JAX and Haiku
000
Finetune_GPTNEO_GPTJ6B
Repo for fine-tuning GPTJ and other GPT models
Language:PythonAGPL-3.0000
Model parallel transformers in JAX and Haiku
Repo for fine-tuning GPTJ and other GPT models