Richard Wong a817fe16cc | ||
---|---|---|
learn_flax | ||
parallel | ||
t5_model | ||
.gitignore | ||
check_time.py | ||
requirements.yaml | ||
t5_jax.py | ||
t5_jax_parallel.py | ||
t5_jax_prediction.py | ||
t5_jax_retrieval.py | ||
t5_jax_sfp_grad_accumulate.py | ||
t5_jax_shmap.py | ||
t5_jax_simple_parallel.py | ||
t5_prediction_old.py | ||
t5_summarizer_flax.py |