richard
/
learn_jax
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
repo tracking my jax experiments
3 Commits 1 Branch 0 Tags 176 KiB
Python 100%
Go to file
Richard Wong d2dd72227f Feat: introduced efficient train data dtype, jit train step, bfloat16 
mat mul
 		2024-09-14 02:02:45 +09:00
.gitignore Feat: introduced efficient train data dtype, jit train step, bfloat16 2024-09-14 02:02:45 +09:00
check_time.py Feat: implement working prediction 2024-09-12 22:57:19 +09:00
requirements.yaml Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00
t5_jax.py Feat: introduced efficient train data dtype, jit train step, bfloat16 2024-09-14 02:02:45 +09:00
t5_jax_prediction.py Feat: introduced efficient train data dtype, jit train step, bfloat16 2024-09-14 02:02:45 +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