-
-
Notifications
You must be signed in to change notification settings - Fork 987
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.
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
Fix TraceEnum_ELBO jit tests, etc. #1590
Conversation
Looks like some JIT-related test failures in |
Hmm weird, the failures only appear in @dgasmith any chance you could cut a minor |
Writing the release notes now. Should cut tomorrow. |
|
Addresses #1418
packed.pack()
inignore_jit_warnings
einsum
when not necessary, usingtensordot
insteadeinsum
patch to allow multiple*operands
(high priority fix!)pyro.ops.einsum.paths
objective function to not break whenopt_einsum
releases their next releaseNote: although the jit tests are still marked xfail, they should start to pass as soon as we switch to the upcoming
opt_einsum
2.3 release.