learn_jax/.gitignore

*.ipynb
model_checkpoints/
exports/
traces/
ruff.toml
settings.json
__pycache__/
Feat: jax implementation of t5 training and prediction 2024-09-11 08:17:02 +09:00			`*.ipynb`
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: 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`
Chore: removing old files and some experiments 2024-10-06 23:53:57 +09:00			`__pycache__/`