strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] linalg_eigh batch rule

zou3519 committed 2 years ago

5acd1b99

[functorch] Fix some headers

zou3519 committed 2 years ago

75a64bb0

[functorch] Implement more batch rules for resnet18 per-sample-grads

zou3519 committed 2 years ago

79e42fc1

[functorch] Fix build/test

zou3519 committed 2 years ago

a054adeb

[functorch] cudnn_convolution decomposition

zou3519 committed 2 years ago

dc5e0c0f

[functorch] maxpool_2d_with_indices_backward batch rule for specific case

zou3519 committed 2 years ago

eb788ae9

[functorch] Implement per sample grad rule for cudnn_convolution_backward

zou3519 committed 2 years ago

1262034b

[functorch] quick fix

zou3519 committed 2 years ago

cfa9d984

[functorch] Introduce gen_plumbing.py

zou3519 committed 2 years ago

4a045e96

[functorch] Very experimental PyTorch forward-mode AD support

zou3519 committed 2 years ago

03fbb542

[functorch] Setup circleci (pytorch/functorch#53)

zou3519 committed 2 years ago

e8c5f67c

[functorch] Added enough things into denylists so that tests pass

zou3519 committed 2 years ago

4aea8064

[functorch] Create "lagging op database", use it in our OpInfo tests

zou3519 committed 2 years ago

3524505d

[functorch] Fix a lot of warnings; use c10::irange

zou3519 committed 2 years ago

fee49501

[functorch] Change initializer list tuple return to std::make_tuple

zou3519 committed 2 years ago

e3429647

[functorch] [BC-breaking] Update make_functional* (pytorch/functorch#52)

zou3519 committed 2 years ago

b29e666a

[functorch] Update tests and examples to use make_functional*_v2

zou3519 committed 2 years ago

fdcc680c

[functorch] Added some notes about pythonkey tracing to readme

Chillee committed 2 years ago

ba1952c1

[functorch] Added std batching rule

Chillee committed 2 years ago

58e7df77

[functorch] templated out sum/mean/var batching rules and added nansum

Chillee committed 2 years ago

710d06c8

[functorch] make_functional*_v2

zou3519 committed 2 years ago

1b90b429

[functorch] vmap-of-vjp and vjp-of-vmap OpInfo testing

zou3519 committed 2 years ago

4a20c215

[functorch] Add newlines to eof

zou3519 committed 2 years ago

e58d7dde

[functorch] Implement batch norm batch rule for one case (where everything is batched)

zou3519 committed 2 years ago

324f4d5e

[functorch] adaptive_avg_pool2d batch rule

zou3519 committed 2 years ago

7a1ee75f

[functorch] Fix nightly binary links

zou3519 committed 2 years ago

804a901a

[functorch] refactored argmax batching rule to support argmin too

Chillee committed 2 years ago

c91ef2f1

[functorch] Fixed argmax batching rules

Chillee committed 2 years ago

039f98d0

[functorch] Added a bunch of unary activation functions

Chillee committed 2 years ago

96444368

[functorch] fix bug with repeat batching rules, fixes pytorch/functorch#9

Chillee committed 2 years ago

f536c20b

Newer Older