flax
Use clu.deterministic_data to shard dataset across hosts.
#815
Merged

Loading