NVIDIA / JAX-Toolbox

JAX-Toolbox
Apache License 2.0
241 stars 45 forks source link

GPT-J #16

Open mjsML opened 1 year ago

mjsML commented 1 year ago

https://github.com/kingoflolz/mesh-transformer-jax