strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] binary_cross_entropy batch rule (pytorch/functorch#380)

zou3519 committed 2 years ago

c6a7b94a

[functorch] fix test failures

Chillee committed 2 years ago

bdcc5286

[functorch] fix flake8 issue

Chillee committed 2 years ago

bbf2e80e

[functorch] fix CI test for an op I don't have locally yet :)

Chillee committed 2 years ago

de4b1e2f

[functorch] fix some random minifier and add some random decompositions

Chillee committed 2 years ago

e47b8f33

[functorch] switch per sample grad test to use group norm

Sam Andow committed 2 years ago

e735de57

[functorch] Add Batch Norm (+ backward) batching rules (pytorch/functorch#371)

Samantha Andow committed 2 years ago

ace198ee

[functorch] Fixed errors

Chillee committed 2 years ago

33110e8a

[functorch] fix flake issues

Chillee committed 2 years ago

747c2fa2

[functorch] refactored decompositions a bit

Chillee committed 2 years ago

02ae449c

[functorch] max-flow based algorithm for activation checkpointing (pytorch/functorch#344)

Chillee committed 2 years ago

3b8b72a3

[functorch] made tests that work on either cpu or cuda but not both skips

Sam Andow committed 2 years ago

a8e01d4b

[functorch] fix ci, forgotten commas and use xfail

Sam Andow committed 2 years ago

dd0fe720

[functorch] fix ci by disabling certain tests

Sam Andow committed 2 years ago

761c3825

[functorch] Dropped 3.6 and added 3.9

vfdev-5 committed 2 years ago

0d37b12f

[functorch] Remove make_nnc from test_pythonkey import

zou3519 committed 2 years ago

47f69917

[functorch] Fix flake8

zou3519 committed 2 years ago

67b8a097

[functorch] README.md: don't include prompt in example code snippets (pytorch/functorch#375)

t-vi committed 2 years ago

647b09bf

[functorch] fix flake errors

Chillee committed 2 years ago

2c1ed695

[functorch] port nnc_jit to use AOTAutograd

Chillee committed 2 years ago

16f34150

[functorch] fix flake rules

Chillee committed 2 years ago

e0987b62

[functorch] fix some test failures

Chillee committed 2 years ago

2aa246e5

[functorch] Added better handling for output device inference (i.e. only use when metatracing)

Chillee committed 2 years ago

89d30b29

[functorch] don't use sum for loss, arbitrary args in torch.Timer benchmarks (pytorch/functorch#369)

ngimel committed 2 years ago

a64602af

[functorch] Add Tutorials to docs build (pytorch/functorch#367)

zou3519 committed 2 years ago

e73417d2

[functorch] quick cleanup

zou3519 committed 2 years ago

fd4d82ee

[functorch] Transcribed tutorials to ipynb

zou3519 committed 2 years ago

649ffb02

[functorch] Wrote tutorials (pytorch/functorch#366)

zou3519 committed 2 years ago

51432e33

[functorch] Fix getitem (pytorch/functorch#364)

zou3519 committed 2 years ago

a1b35d83

[functorch] Added cdist forward/backward batching rules (pytorch/functorch#306)

vfdev-5 committed 2 years ago

99b842d6

Newer Older