strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Batching rules for comparison ops

zou3519 committed 2 years ago

cb51b16e

[functorch] vjp now supports pytree inputs and outputs

zou3519 committed 2 years ago

f92fbeef

[functorch] Fix vjp and grad for unrelated outputs

zou3519 committed 2 years ago

9490aa0b

[functorch] link colab with install instructions

zou3519 committed 2 years ago

ec99d21d

[functorch] Use functorch._C._set_vmap_fallback_warning_enabled

zou3519 committed 2 years ago

9a812032

[functorch] Beef up make_functional docstring, update some examples

zou3519 committed 2 years ago

15ab42ce

[functorch] fixed some python key tracing issues

Chillee committed 2 years ago

6ecf169a

[functorch] Removed some stupid stuff

Chillee committed 2 years ago

7224611c

[functorch] adds support for grad if inputs don't depend on output (pytorch/functorch#6)

Chillee committed 2 years ago

80131b93

[functorch] fix vmap tests and added dot shape check

Chillee committed 2 years ago

86e49cf0

[functorch] removed nnc_compile

Chillee committed 2 years ago

73f57a71

[functorch] Added python key stuff that builds with master

Chillee committed 2 years ago

98e995c4

[functorch] merged

Chillee committed 2 years ago

9d36895a

[functorch] Fix up the cifar10_transforms example

zou3519 committed 2 years ago

98df806b

[functorch] pytree output support for vmap

zou3519 committed 2 years ago

dfeb7898

[functorch] Add pytree support for grad and grad_and_value

zou3519 committed 2 years ago

c6773c67

[functorch] Fix build

zou3519 committed 2 years ago

8c8685ca

[functorch] Installation instructions

zou3519 committed 2 years ago

ba91512c

[functorch] Added ones_like/zeros_like/... batching rules

Chillee committed 2 years ago

32240466

[functorch] Added debug commands

Chillee committed 2 years ago

03550ccd

[functorch] removed python key stuff

Chillee committed 2 years ago

033e6e30

[functorch] Quick temporary fix for python key

zou3519 committed 2 years ago

24677739

[functorch] fixed dot batching rule with no batch dims

Chillee committed 2 years ago

c521cd1f

[functorch] fixed dot batching rule and added vmap tests using opinfo

Chillee committed 2 years ago

0d94ae66

[functorch] Added dot/add.Scalar/mul.Scalar/etc. batching rules and added functools.wraps to grad

Chillee committed 2 years ago

918ede7a

[functorch] modified example

Chillee committed 2 years ago

ac9be17a

[functorch] More batch rules

zou3519 committed 2 years ago

c24314c0

[functorch] Grab bag of batch rules

zou3519 committed 2 years ago

d7d266f5

[functorch] fix testing warning, update readme

zou3519 committed 2 years ago

0abba43a

[functorch] Add codegen file; beef up testing

zou3519 committed 2 years ago

42843306

Newer Older