Richard Wong aca80720c8 | ||
---|---|---|
.. | ||
.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 |