strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Added a simple function example

Chillee committed 2 years ago

ac017463

[functorch] Added nnc compilation stuff to functorch

Chillee committed 2 years ago

25453d61

[functorch] Migrate mm and mv batch rules from old style to new style

zou3519 committed 2 years ago

4b0d62a2

[functorch] reshape_dim_into and reshape_dim_outof helpers

zou3519 committed 2 years ago

9bdd9cee

[functorch] Delete some redundant code

zou3519 committed 2 years ago

0c5755fc

[functorch] Added batching rules for some in-place pointwise ops

zou3519 committed 2 years ago

58ee527d

[functorch] Refactory UnaryOps

zou3519 committed 2 years ago

66918c4a

[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

Newer Older