strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] Run all correctness tests

zou3519 committed 2 years ago

e8abf483

[functorch] testing, version.txt

zou3519 committed 2 years ago

20fac9da

[functorch] Update readme, fix unsqueeze batch rule

zou3519 committed 2 years ago

8277d74e

[functorch] Update README, setup.py

zou3519 committed 2 years ago

7001a2f1

[functorch] Update

zou3519 committed 2 years ago

70925e47

[functorch] misc cleanup

zou3519 committed 2 years ago

b7096ab8

[functorch] Updated some batching rules to use new API

zou3519 committed 2 years ago

ce453d44

Newer Older