strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Nested jvp seems to work

zou3519 committed 2 years ago

1c96132a

[functorch] Some view backward batch rules

zou3519 committed 2 years ago

0753d0c4

[functorch] Revert "some view backward batch rules"

zou3519 committed 2 years ago

839517a3

[functorch] Fix README typo. (pytorch/functorch#120)

dan-zheng committed 2 years ago

ae1c2047

[functorch] some view backward batch rules

zou3519 committed 2 years ago

b6412da1

[functorch] Kill SkipInfo

zou3519 committed 2 years ago

0db39623

[functorch] remove scatter expected failure

Sam Andow committed 2 years ago

b47bc8f0

[functorch] move select to new API

Sam Andow committed 2 years ago

975a2df0

[functorch] Added VMAP_SUPPORT for returning vectors of tensors

Chillee committed 2 years ago

9cba6926

[functorch] Added wraps(f) to jacrev

Chillee committed 2 years ago

3a49b986

[functorch] removed some unnecessary batching rules from batchingregistrations.cpp

Chillee committed 2 years ago

dc9cbe1d

[functorch] Adding jacrev numargs argument (pytorch/functorch#111)

Samantha Andow committed 2 years ago

f20150a3

[functorch] removed unnecessary note

Chillee committed 2 years ago

b23c9b90

[functorch] removed stupid line

Chillee committed 2 years ago

fcd08e1a

[functorch] Fix location of file

zou3519 committed 2 years ago

ae5a1290

[functorch] Fix bug in discover_coverage

zou3519 committed 2 years ago

8bd71e0a

[functorch] Handle label smoothing in cross_entropy

zou3519 committed 2 years ago

818487a5

[functorch] Added dlpack conversion to/from for TVM

Chillee committed 2 years ago

7af7eddf

[functorch] Added handling of parameters so that requires_grad is propagated correctly

Chillee committed 2 years ago

9168d834

[functorch] merged

Chillee committed 2 years ago

b1390744

[functorch] added option to skip specialization cache

Chillee committed 2 years ago

7e13e3c6

[functorch] Cleaned up the tests

Chillee committed 2 years ago

c0afd464

[functorch] fix some tests

Chillee committed 2 years ago

0562725c

[functorch] stored some stuff

Chillee committed 2 years ago

d314e2f7

[functorch] moved function to eager_compilation

Chillee committed 2 years ago

cb22fefe

[functorch] Added start of compile_funciton opinfo tests

Chillee committed 2 years ago

09b23abf

[functorch] Added static_argnums to nnc_jit

Chillee committed 2 years ago

4ff4501d

[functorch] actually fixed tests?

Chillee committed 2 years ago

dcea4261

[functorch] fix warnings

Chillee committed 2 years ago

66586e95

[functorch] moved a bunch of inplace ops to use boxed fallback

Chillee committed 2 years ago

a528be7d

Newer Older