2 ms·
You may already know of it, but if you want differential-equations-in-JAX then allow me to quickly advertise Diffrax: https://github.com/patrick-kidger/diffrax
by patrickkidger 4y ago
You may already know of it, but if you want differential-equations-in-JAX then allow me to quickly advertise Diffrax: https://github.com/patrick-kidger/diffrax https://github.com/patrick-kidger/diffrax (of which I am the author, disclaimer).
- orbifold 4y agoYes I am aware :) it is missing a few things but I might end up contributing.
- patrickkidger 4y agoExcellent! I'm very happy to take contributions generalising the tool.