learn_jax/.gitignore

*.ipynb
t5_*/
exports/
modified_t5_model/
traces/
Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00			`*.ipynb`
			`t5_*/`
			`exports/`
Feat: implement working prediction 2024-09-12 22:57:19 +09:00			`modified_t5_model/`
Feat: introduced efficient train data dtype, jit train step, bfloat16 mat mul 2024-09-14 02:02:45 +09:00			`traces/`