Patrick Kidger

So what's going on here is that the previously-static leaves are array-like, so they get selected for tracing by `filter_jit(filter_spec=is_array_like)`. As a result they are converted into JAX arrays. Meanwhile...

Which tests actually break under this change? (+with what error if it's interesting.) Also, you might try running the tests for Diffrax under that change and see if any of...

Don't worry, it definitely isn't going away. If nothing else it's a key part of how things work internally. On balance I think I'm inclined to maintain the status quo....

Have a look at the latent ODE and neural CDE examples in the documentation.

Closing in favour of #181.

Sounds good to me. As usual, I'd be happy to accept a PR on this.

Accomplished in #176.

Hey there! Thanks for your interest. I think we clearly have the same conceptual preferences: using `__init__` rather than a new classmethod was important to me to. Have you seen...

I'm glad to hear it!

I'm afraid your code isn't something that can be run standalone -- if you can try to provide a snippet of code that I can copy-paste to understand what you're...