Brett M. Morris
Brett M. Morris
I can confirm the fix above works for me for now. Thanks Dan!
Hi @dfm, sorry to be plaguing you 😅. I'm working on a JAX project with GPU acceleration and I'd like to use celerite2. If I use it out of the...
Thanks for the quick response! If you have any pointers on alternatives I'd be grateful.
Thanks so much, as always!
This probably falls under the "Traced value used in control flow" section of the [docs for `ConcretizationTypeError`](https://jax.readthedocs.io/en/latest/errors.html). JAX can't differentiate over that `if`.
Hi @dfm! I'm starting to work with the jax backend, and I'm hitting an error: ``` NotImplementedError: Batching rule for 'celerite2_solve_lower_jvp' not implemented ``` Is that what you're referring to...
Ah I see this was because I left my `NUTS(...,forward_mode_differentiation=True)`, sorry for the noise!
Those static pages are beautiful, that's awesome @bsipocz! The main consideration is that the navo workshop uses [jupytext](https://github.com/mwouts/jupytext) to convert between markdown files and Jupyter notebooks. This adds a bit...
You said you're running Windows 10, so Windows.
Sorry, I've been corrected offline! If you're using WSL, you likely need the Linux installation. Let me know if that works.