Richard Wong 7e1f45f466 | ||
---|---|---|
.. | ||
t5_model | ||
.gitignore | ||
dataload.py | ||
flax_pjit_tutorial.py | ||
fully_sharded_data_parallelism.py | ||
gpt-neo-125m.json | ||
gptneo_partition_test.py | ||
intro_to_distributed.py | ||
partitions.py | ||
single_gpu_optimizations.py | ||
t5.json | ||
t5_jax_train_2.py | ||
t5_jax_train_fail.py | ||
t5_pjit.py |