transformers
Speedup T5 Flax training by using Numpy instead of JAX for batch shuffling
#15963

Merged

Speedup T5 Flax training by using Numpy instead of JAX for batch shuffling #15963

patil-suraj merged 1 commit into huggingface:master from yhavinga:speedup-train-with-numpy-batch-shuffling

Speedup training by using numpy instead of jnp for batch shuffling

a982162b

patrickvonplaten approved these changes on 2022-03-07

patrickvonplaten requested a review from

patil-suraj 4 years ago

patil-suraj approved these changes on 2022-03-08

patil-suraj merged 91fb62d0 into master 4 years ago

Reviewers

patil-suraj

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone