What's Changed
* Fix handling of `in_axes` and out_axes` in internal used of `filter_vmap` by vadmbertr in https://github.com/vadmbertr/pastax/pull/20
* Performance tricks by vadmbertr in https://github.com/vadmbertr/pastax/pull/21
* replace arguments values : `dt0=None` and `saveat=dfx.SaveAt(ts=ts)`
* add argument `stepsize_controller=dfx.StepTo(...)`
* Stochastic dynamics now return a lineax.PyTreeLinearOperator and SDEControl.__call__ a corresponding valid Pytree vadmbertr in https://github.com/vadmbertr/pastax/pull/21
* Several small typings refactoring vadmbertr in https://github.com/vadmbertr/pastax/pull/21
* Update the last part of the `getting_started` notebook vadmbertr in https://github.com/vadmbertr/pastax/pull/21
* Improves how `diffeqsolve` arguments are derived from arguments passed to `DiffraxSimulator.__call__` by vadmbertr in https://github.com/vadmbertr/pastax/pull/22 (changes introduced in https://github.com/vadmbertr/pastax/pull/21)
* Fix docs dependencies version by vadmbertr in https://github.com/vadmbertr/pastax/pull/23
**Full Changelog**: https://github.com/vadmbertr/pastax/compare/0.0.3...0.0.4