strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Port jvp decompositions to unify on python one (pytorch/functorch#772)

Chillee committed 2 years ago

9a6b91b6

[functorch] Revert "Port nll_loss_backward decomposition to unify on python one"

Chillee committed 2 years ago

c0707c6f

[functorch] Port nll_loss_backward decomposition to unify on python one

Chillee committed 2 years ago

b208b960

[functorch] Add trace decomposition, invoke from C++ (pytorch/functorch#740)

eellison committed 2 years ago

3cd45f7c

[functorch] Minor fix (pytorch/functorch#769)

cyyever committed 2 years ago

57cf01df

[functorch] [jvp x vjp] remove conv skips

zou3519 committed 2 years ago

5318b616

[functorch] [jvp x vjp] cross entropy

zou3519 committed 2 years ago

fe203ad1

[functorch] fixup jvpvjp testing, add log_softmax and softmax jvpvjp coverage

zou3519 committed 2 years ago

7314fd24

[functorch] jvp x vjp support for l1_loss, mse_loss

zou3519 committed 2 years ago

f10e9d66

[functorch] fix thing I thought I already fixed

Chillee committed 2 years ago

b51e9b50

[functorch] fixed torchscriptable error

Chillee committed 2 years ago

ce6a8057

[functorch] Use nll_loss_backward decomposition for jvp transform (pytorch/functorch#764)

zou3519 committed 2 years ago

babc862b

[functorch] Update discover_coverage, add jvp support for linear

zou3519 committed 2 years ago

41869d9a

[functorch] Fix a bunch of linalg coverage issuez (pytorch/functorch#765)

zou3519 committed 2 years ago

0f1332b0

[functorch] Fix some minor issues (pytorch/functorch#768)

cyyever committed 2 years ago

42286432

[functorch] fixed rsub

Chillee committed 2 years ago

ec21c908

[functorch] fix flake

Chillee committed 2 years ago

99abac4b

[functorch] fixed complex number decompositions and added isnan

Chillee committed 2 years ago

855f939c

[functorch] fix CI

Chillee committed 2 years ago

aba4a67b

[functorch] Added batch norm decomposition and fixed some other stuff

Chillee committed 2 years ago

3dccff78

[functorch] Added clone implement for ProxyTensor __deepcopy__

Chillee committed 2 years ago

6f7ea6d8

[functorch] Some more test annotations

zou3519 committed 2 years ago

01164988

[functorch] re-annotated some tests

zou3519 committed 2 years ago

e86d8ebf

[functorch] Refactor DynamicLayer so the logic for each transform is separate (pytorch/functorch#756)

zou3519 committed 2 years ago

3bf44893

[functorch] Add xfails for nansum, nanmean

zou3519 committed 2 years ago

8a6491c8

[functorch] Only generate vmap plumbing for operations in the allowlist (pytorch/functorch#758)

zou3519 committed 2 years ago

2788574f

[functorch] nll loss decomp (pytorch/functorch#755)

Samantha Andow committed 2 years ago

60214759

[functorch] L1 loss decomposition (pytorch/functorch#753)

Samantha Andow committed 2 years ago

b2b1a00d

[functorch] fix binary_cross_entropy_backward decomposition and fix _tensor_str

Chillee committed 2 years ago

ac9a47bf

[functorch] xfail binary cross entropy decomposition on cuda, bfloat16

samdow committed 2 years ago

fd05b533

Newer Older