repo tracking my jax experiments
Go to file
Richard Wong f523560141 Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
.gitignore Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
requirements.yaml Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
t5_jax.py Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
t5_jax_prediction.py Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
t5_jax_retrieval.py Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
t5_summarizer_flax.py Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00