strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Update jacrev to accept functions with pytree inputs and outputs (pytorch/functorch#297)

zou3519 committed 2 years ago

b912da12

[functorch] [batching] select_scatter (pytorch/functorch#289)

kshitij12345 committed 2 years ago

3deb38e6

[functorch] [batching] isinf, isreal (pytorch/functorch#294)

kshitij12345 committed 2 years ago

70e052a8

[functorch] layer_norm batch rule (pytorch/functorch#293)

zou3519 committed 2 years ago

7f0d0fe5

[functorch] Fix CI

zou3519 committed 2 years ago

5daececc

[functorch] Fix functorch additional ops db

zou3519 committed 2 years ago

e7b95307

[functorch] OutOfPlacePlumbing now gets annotated with aten ops

zou3519 committed 2 years ago

1b33ea0d

[functorch] [batching] index_copy (pytorch/functorch#282)

kshitij12345 committed 2 years ago

fada7dd5

[functorch] add missing undef (pytorch/functorch#291)

kshitij12345 committed 2 years ago

1a6d3cd0

[functorch] Added logical_and/or/xor vmaps to enable torch.pow (pytorch/functorch#286)

vfdev-5 committed 2 years ago

7506e8b8

[functorch] Fix method rename (pytorch/functorch#288)

vfdev-5 committed 2 years ago

ecf5817a

[functorch] Added adaptive_max_poolNd batch rule (pytorch/functorch#263)

vfdev-5 committed 2 years ago

99ed1b27

[functorch] Removed some xfail for nn functional pad circular (pytorch/functorch#285)

vfdev-5 committed 2 years ago

1f9c78cc

[functorch] copy batch rule (pytorch/functorch#278)

zou3519 committed 2 years ago

9e6a6f5d

[functorch] new_blah batch rules (pytorch/functorch#277)

zou3519 committed 2 years ago

907fc882

[functorch] cross_entropy_loss decomposition, fixes to coverage detection (pytorch/functorch#276)

zou3519 committed 2 years ago

461e779c

[functorch] improve coverage testing

zou3519 committed 2 years ago

4e3b1503

[functorch] hessian API

zou3519 committed 2 years ago

cdb56a0e

[functorch] exported register_decomposition to be public

Chillee committed 2 years ago

4e75634a

[functorch] Fix edge case with in-place captures and grad transforms (pytorch/functorch#272)

zou3519 committed 2 years ago

f1c9426c

[functorch] Revert "Making some things kwarg only"

zou3519 committed 2 years ago

21a4c9fb

[functorch] Making some things kwarg only

zou3519 committed 2 years ago

37123cd2

[functorch] Strict mode for jvp

zou3519 committed 2 years ago

30f787c6

[functorch] Fix unrelated output behavior for jvp

zou3519 committed 2 years ago

b8e25d25

[functorch] jacfwd supports multiple inputs and multiple outputs (pytorch/functorch#270)

zou3519 committed 2 years ago

1b10fb4f

[functorch] jvp transform accepts functions that take multiple inputs and outputs (pytorch/functorch#267)

zou3519 committed 2 years ago

e19546fb

[functorch] jacrev, multiple outputs support (pytorch/functorch#265)

zou3519 committed 2 years ago

403e2962

[functorch] Added im2col batch rule and enabled vmap for nn.functional.unfold op (pytorch/functorch#262)

vfdev-5 committed 2 years ago

5a58d5f7

[functorch] changed naming from aot autograd to eager compilation

Chillee committed 2 years ago

8769c9a2

[functorch] updated some eager compilation code

Chillee committed 2 years ago

1eaf24c9

Newer Older