learn_jax/.gitignore

*.ipynb
t5_*/
model_checkpoints/
exports/
modified_t5_model/
traces/
ruff.toml
settings.json
Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00			`*.ipynb`
			`t5_*/`
Feat: t5_jax_simple_parallel implements a working example of fsdp 2024-09-20 23:42:51 +09:00			`model_checkpoints/`
Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00			`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/`
Feat: learn flax 2024-09-14 14:13:38 +09:00			`ruff.toml`
			`settings.json`