Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reproduce DR learner notebook using chirho.robust #474

Merged
merged 88 commits into from
Jul 18, 2024
Merged

Reproduce DR learner notebook using chirho.robust #474

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
88 commits
Select commit Hold shift + click to select a range
38b6158
added robust folder
agrawalraj Nov 8, 2023
c09dcab
uncommited scratch work for log prob
agrawalraj Nov 9, 2023
21e31bf
untested variational log prob
agrawalraj Nov 9, 2023
faed235
uncomitted changes
agrawalraj Nov 13, 2023
fac98cd
uncomitted changes
agrawalraj Nov 16, 2023
4edcb5e
pair coding w/ eli
agrawalraj Nov 16, 2023
fe17403
added tests w/ Eli
agrawalraj Nov 17, 2023
b159687
eif
eb8680 Nov 17, 2023
33f4811
linting
agrawalraj Nov 18, 2023
8e171f4
moving test autograd to internals and deleted old utils file
agrawalraj Nov 20, 2023
93cc014
sketch influence implementation
eb8680 Nov 21, 2023
9bc704c
fix more args
eb8680 Nov 21, 2023
cedb818
ops file
eb8680 Nov 21, 2023
418f792
file
eb8680 Nov 21, 2023
f792ddf
format
eb8680 Nov 21, 2023
88a100b
lint
eb8680 Nov 21, 2023
94c2fc6
clean up influence and tests
eb8680 Nov 21, 2023
da0bc5c
make tests more generic
eb8680 Nov 22, 2023
4d027e4
guess max plate nesting
eb8680 Nov 22, 2023
e85e33f
linearize
eb8680 Nov 22, 2023
1734191
rename file
eb8680 Nov 22, 2023
f46556b
tensor flatten
eb8680 Nov 22, 2023
1abc5e0
predictive eif
eb8680 Nov 22, 2023
9c80b60
jvp type
eb8680 Nov 22, 2023
931da4f
reorganize files
eb8680 Nov 22, 2023
dc63f31
shrink test case
eb8680 Nov 22, 2023
be3bc8d
move guess_max_plate_nesting
eb8680 Nov 22, 2023
9ce164a
move cg solver to linearze
eb8680 Nov 22, 2023
81196d4
type alias
eb8680 Nov 22, 2023
30cb2e7
test_ops
eb8680 Nov 22, 2023
21cf2d7
basic cg tests
eb8680 Nov 22, 2023
720661f
remove failing test case
eb8680 Nov 22, 2023
91833da
format
eb8680 Nov 22, 2023
548069a
move paramdict up
eb8680 Nov 22, 2023
12b22c0
remove obsolete test files
eb8680 Nov 22, 2023
d2bbf9d
Merge branch 'master' into staging-robust
eb8680 Nov 22, 2023
3b72bb0
add empty handlers
eb8680 Nov 22, 2023
89d9f6b
add chirho.robust to docs
eb8680 Nov 22, 2023
7582c22
fix memory leak in tests
eb8680 Nov 27, 2023
82c23e8
make typing compatible with python 3.8
eb8680 Nov 27, 2023
e08d9d6
typing_extensions
eb8680 Nov 27, 2023
22eae09
add branch to ci
eb8680 Nov 27, 2023
d0014db
predictive
eb8680 Nov 27, 2023
e5342dc
remove imprecise annotation
eb8680 Nov 27, 2023
be13ac5
Merge branch 'master' into staging-robust
SamWitty Nov 28, 2023
c5fe64b
Added more tests for `linearize` and `make_empirical_fisher_vp` (#405)
agrawalraj Dec 6, 2023
117d645
Add upper bound on number of CG steps (#404)
eb8680 Dec 7, 2023
8fe1b25
fixed test for non-symmetric matrix (#437)
agrawalraj Dec 7, 2023
3f0c83d
Make `NMCLogPredictiveLikelihood` seeded (#408)
agrawalraj Dec 8, 2023
4d41807
Use Hessian formulation of Fisher information in `make_empirical_fish…
agrawalraj Dec 8, 2023
2e01b7b
Add new `SimpleModel` and `SimpleGuide` (#440)
agrawalraj Dec 8, 2023
538cef8
Batching in `linearize` and `influence` (#465)
agrawalraj Dec 22, 2023
6bba70b
batched cg (#466)
agrawalraj Dec 22, 2023
f143d3a
One step correction implemented (#467)
agrawalraj Dec 22, 2023
878eb0d
Replace some `torch.vmap` usage with a hand-vectorized `BatchedNMCLog…
eb8680 Jan 2, 2024
3cfe319
Added documentation for `chirho.robust` (#470)
agrawalraj Jan 2, 2024
78ed274
old dr notebook that got deleted from wrong merge
agrawalraj Jan 2, 2024
6440ada
added missing fig
agrawalraj Jan 2, 2024
6daea7f
redid notebook with new interface
agrawalraj Jan 2, 2024
5d77fe0
Make functional argument to influence_fn required (#487)
eb8680 Jan 9, 2024
013d518
Remove guide argument from `influence_fn` and `linearize` (#489)
eb8680 Jan 9, 2024
7eeb941
updated labels
agrawalraj Jan 10, 2024
80b1678
Merge branch 'staging-robust' of github.com:BasisResearch/causal_pyro…
agrawalraj Jan 10, 2024
f326ef3
updated w/ new interface but only 1 data sim
agrawalraj Jan 10, 2024
c4346c8
Make influence_fn a higher-order Functional (#492)
eb8680 Jan 11, 2024
e43f5f7
uncommitted changes
agrawalraj Jan 11, 2024
9207e3e
Add full corrected one step estimator (#476)
SamWitty Jan 12, 2024
2ac0e40
kernel speedup
agrawalraj Jan 12, 2024
ca916cd
Merge branch 'master' into staging-robust
eb8680 Jan 12, 2024
46ad342
before switching to krr formulation
agrawalraj Jan 13, 2024
ff80d8d
uncommitted changes
agrawalraj Jan 15, 2024
0f7165e
Merge branch 'staging-robust' of github.com:BasisResearch/causal_pyro…
agrawalraj Jan 15, 2024
1bdf9f0
updated w/ new interface; removed GP section for now
agrawalraj Jan 15, 2024
bf6718d
runs but not matching
agrawalraj Jan 15, 2024
f04e35b
still not working, going to make major changes
agrawalraj Jan 16, 2024
e09bfb3
Merge branch 'master' into ra-dr-notebook2
eb8680 Jul 10, 2024
ca3d70f
remove debug script
eb8680 Jul 10, 2024
832dc7c
remove file
eb8680 Jul 10, 2024
945cded
remove file
eb8680 Jul 10, 2024
8771cdb
add
eb8680 Jul 10, 2024
ac8ccd9
update interfaces
eb8680 Jul 10, 2024
49c5d59
finished running
eb8680 Jul 10, 2024
c8c9144
Merge branch 'master' into ra-dr-notebook2
eb8680 Jul 16, 2024
3547ce5
outline
eb8680 Jul 16, 2024
93fead5
remove outline for now
eb8680 Jul 16, 2024
a9a4419
simplify notebook
eb8680 Jul 18, 2024
b76606e
Merge branch 'master' into ra-dr-notebook2
eb8680 Jul 18, 2024
1dc594f
merge
eb8680 Jul 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading