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

Jump to bottom

Gradients through FoM fails when using cubic spline interpolation #57

Open
EiffL opened this issue Jul 21, 2020 · 0 comments · May be fixed by #104
Open

Gradients through FoM fails when using cubic spline interpolation #57

EiffL opened this issue Jul 21, 2020 · 0 comments · May be fixed by #104
Assignees
@EiffL
Labels
bug Something isn't working JAX Issues related to JAX

Comments

@EiffL
Copy link
Member

EiffL commented Jul 21, 2020

For some reason, gradients fail with a cryptic:

<class 'jax.ad_util.Zero'> is not a valid JAX type

This can be replicated for instance when using this notebook:
https://github.com/EiffL/tomo_challenge/blob/master/notebooks/LearningToBin-2bins-FoM.ipynb
with the code from this branch: u/EiffL/spline_interpolation
@EiffL EiffL added bug Something isn't working JAX Issues related to JAX labels Jul 21, 2020
@EiffL EiffL self-assigned this Jul 21, 2020
@EiffL EiffL linked a pull request Dec 21, 2022 that will close this issue
Switching to integration via Splines #104
Draft
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@EiffL EiffL

Labels
bug Something isn't working JAX Issues related to JAX
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Switching to integration via Splines DifferentiableUniverseInitiative/jax_cosmo
1 participant
@EiffL