GPT-J-6B
https://github.com/kingoflolz/mesh-transformer-jax?tab=readme-ov-file#gpt-j-6b
#JAX
A 6 billion parameter, autoregressive text generation model trained on
The Pile
.