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.

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 ProductBijector on Julia <1.9 #314

Merged
merged 2 commits into from
Jun 3, 2024
Merged

Fix ProductBijector on Julia <1.9 #314

merged 2 commits into from
Jun 3, 2024

Conversation

torfjelde
Copy link
Member

@torfjelde torfjelde changed the base branch from master to patch June 3, 2024 14:02
@torfjelde
Copy link
Member Author

@sethaxen Tests are failing because of numerical issues with cholesky transforms, so I'm making the PR directly to your branch instead. Happy with this?

@sethaxen
Copy link
Member

sethaxen commented Jun 3, 2024

Sure that's fine

@yebai yebai merged commit 21189fa into patch Jun 3, 2024
15 of 22 checks passed
@delete-merged-branch delete-merged-branch bot deleted the torfjelde/stack-fix branch June 3, 2024 22:40
torfjelde added a commit that referenced this pull request Jun 5, 2024
* Rename VecCholeskyBijector to VecCorrCholeskyBijector

* Compute corr logdetjac during transform

* Enforce one-based indexing

* Add with_logabsdet_jacobian for correlation transforms

* Add rrule for non-mutating ADs

* Update ChainRules to use manual rrule

* Update Tracker to use manual rrule

* Remove rrule for ReverseDiff

`@grad_from_chainrules` can't handle multi-output functions, see JuliaDiff/ReverseDiff.jl#221. In this case it can AD through the primal just fine.

* Add module

* Make CorrBijector more numerically stable

Also use consistent notation with inverse transform

* Increment patch number

* Revert "Rename VecCholeskyBijector to VecCorrCholeskyBijector"

This reverts commit bd6ff3d.

* Update src/bijectors/corr.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Tor Erlend Fjelde <tor.erlend95@gmail.com>

* Apply suggestions from code review

* Work around issues with Tracker

* import `stack` from Compat.jl (#314)

* import `stack` in tests too

* disable certain tests for ProductBijector on Julia versions with older
`eachslice` impls

---------

Co-authored-by: Tor Erlend Fjelde <tor.erlend95@gmail.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants