transformers
Support compilation via Torchdynamo, AOT Autograd, NVFuser
#17308
Merged

Support compilation via Torchdynamo, AOT Autograd, NVFuser #17308

sgugger merged 10 commits into huggingface:main from anijain2305:use_dynamo
anijain2305
anijain2305 Support compilation via Torchdynamo, AOT Autograd, NVFuser
23962182
anijain2305 anijain2305 force pushed to 23962182 3 years ago
HuggingFaceDocBuilderDev
stas00 stas00 assigned stas00 stas00 3 years ago
stas00 stas00 added Performance
stas00
stas00 requested changes on 2022-05-17
sgugger
sgugger commented on 2022-05-18
anijain2305 Address comments
0b4c279a
anijain2305 anijain2305 force pushed to 0b4c279a 3 years ago
sgugger
sgugger approved these changes on 2022-05-19
anijain2305 anijain2305 force pushed 3 years ago
anijain2305 Lint
52503f2f
anijain2305 anijain2305 force pushed to 52503f2f 3 years ago
stas00
stas00 requested changes on 2022-05-19
csarofeen
stas00
kevinstephano
anijain2305 Stas comments - missing quality test
298b3baa
anijain2305 Lintere
6756fa53
csarofeen
Chillee
anijain2305 Quality test
28f80ec0
stas00
stas00 commented on 2022-05-19
stas00
Chillee
stas00
Chillee
stas00
anijain2305 Doc lint
bf417042
anijain2305 Reset CUDA peak mem
7925fa33
anijain2305
stas00
anijain2305
anijain2305 commented on 2022-05-25
anijain2305 Add CustomTrainer
ce476195
stas00 require a single gpu
9daaf930
stas00
stas00 approved these changes on 2022-05-25
sgugger sgugger merged 897a8dd8 into main 3 years ago
stas00
anijain2305
stas00
stas00

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone