PR #23168 shift torch dynamo handling to accelerate

shift torch dynamo handling to accelerate #23168

pacman100 merged 15 commits into main from smangrul/accelerate-dynamo-integrate

mixed precision support via accelerate

b3987a83

fix issues

862d04bb

fix for the sharded ddp case

f2196bec

fix flax and tf failing tests

2339a485

`refactor the place to create `Accelerator` object

263b1343

move ddp prep to accelerate

a5bf5175

fix 😅

f00ce091

resolving comments

254f9a46

move fsdp handling to accelerate

88e73508

fixex

b37ad2ad

fix saving

ec73bf22

shift torch dynamo handling to accelerate

ed1a5203

pacman100 requested a review from

sgugger 2 years ago

pacman100 requested a review from

muellerzr 2 years ago

muellerzr approved these changes on 2023-05-05

sgugger approved these changes on 2023-05-05

Merge branch 'main' into smangrul/accelerate-dynamo-integrate

b2d99461

pacman100 changed the base branch from smangrul/accelerate-fsdp-integrate to main 2 years ago

pacman100 changed the base branch from main to smangrul/accelerate-fsdp-integrate 2 years ago

Base automatically changed from smangrul/accelerate-fsdp-integrate to main 2 years ago

Merge branch 'main' into smangrul/accelerate-dynamo-integrate

834ec383

Merge branch 'main' into smangrul/accelerate-dynamo-integrate

53676448

pacman100 merged 03db5910 into main 2 years ago

pacman100 deleted the smangrul/accelerate-dynamo-integrate branch 2 years ago

Reviewers

sgugger

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers shift torch dynamo handling to accelerate #23168 Merged

shift torch dynamo handling to accelerate #23168

transformers
shift torch dynamo handling to accelerate
#23168

Merged