PR #17760 Flax sharded - SemanticDiff

Flax sharded #17760

ArthurZucker merged 18 commits into huggingface:main from ArthurZucker:flax-sharded

initial commit

cc773a2c

style

62df3a17

update

ea656fc6

fix flax test

81e7ab8a

functions in hub

e62d65b8

Update

db435a15

fix bug

2fba0be7

Update and fix test

d631191d

update and fix test, now running

a664889a

ArthurZucker marked this pull request as ready for review 3 years ago

ArthurZucker requested a review from

patil-suraj 3 years ago

ArthurZucker requested a review from

sgugger 3 years ago

ArthurZucker requested a review from

gante 3 years ago

ArthurZucker assigned

ArthurZucker 3 years ago

ArthurZucker added Core: Modeling

ArthurZucker removed review request from

gante 3 years ago

ArthurZucker requested a review from

patrickvonplaten 3 years ago

patrickvonplaten requested a review from

sanchit-gandhi 3 years ago

sanchit-gandhi approved these changes on 2022-06-21

patil-suraj approved these changes on 2022-06-21

sgugger approved these changes on 2022-06-21

Apply suggestions from code review

b96949e8

update based on review

24670f06

Merge branch 'flax-sharded' of https://github.com/ArthurZucker/transf…

cd89fa52

style

5f0b8c37

clean

111688fd

remove conversion to np array

83ec3886

Merge branch 'main' of https://github.com/huggingface/transformers in…

1d48275e

Merge branch 'main' of https://github.com/huggingface/transformers in…

128b666a

Merge branch 'main' into flax-sharded

3261abf1

ArthurZucker merged 16c6eb7c into main 3 years ago

Reviewers

sgugger

patil-suraj

sanchit-gandhi

patrickvonplaten

Assignees

ArthurZucker

Labels

Core: Modeling

Milestone

No milestone

transformers Flax sharded #17760 Merged

Flax sharded #17760

transformers
Flax sharded
#17760

Merged