Patrick Kidger
Patrick Kidger
As title. `sigbackprop` and `logsigbackprop` return float32 even when input is float64, and `sig` and `logsig` return float64 if the input is float32. e.g. ```python import numpy as np import...
Fixes #166.
This looks like it might be a bit of fun to try when displaying PyTrees i.e. models: https://github.com/AharonSambol/PrettyPrintTree
TL;DR: XLA sub-graphs! ### Background At present, JAX inlines the entire computation into a single XLA graph. However, many scientific computing applications involve defining some modestly complicated function and then...
People are starting to build cool stuff with Equinox. It might be worth adding links to these in the documentation for examples. - Diffrax - sympy2jax - Lucidrains' work -...
By adding a few meta tags it's possible to get Twitter to display a custom image etc. when linking to the documentation. Adding this in to the docs sounds like...
At the moment, `get_state` and `set_state` demand that their arguments have precisely the same shape, dtype, and choice of batch axes. It would make sense to allow some compatibility between...
For each adjoint method, it would be convenient to include a table specifying which are compatible with each kind of autodifferentiation.
At the moment it uses `jnp.linalg.solve`, which assumes that the solve operation is well-posed. This needn't actually be the case, though, and at the moment the gradient we obtain could...