patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
PythonApache-2.0
Issues
- 2
Neural ODE Example Fails on Jax v0.4.33
#503 opened by allen-adastra - 3
Value error when used in gradient optimization with equinox latest version
#502 opened by ParticularlyPythonicBS - 7
Why can't UBP be backproped?
#490 opened by lockwo - 2
Parallel Testing
#486 opened by lockwo - 3
Different step sizes within batch?
#500 opened by jucor - 5
Feature Request: Complex-Valued Integration With ZVODE - CVODE in Jax (autodiff)
#477 opened by onurdanaci - 1
Add weak order to solvers?
#497 opened by lockwo - 4
Incorrect gradient in toy adaptive ODE
#499 opened by lockwo - 7
VBT vs brownian path slowdown
#489 opened by lockwo - 1
Levy Area documentation
#491 opened by lockwo - 1
Implement the IAS15 solver
#437 opened by nstarman - 10
- 2
Restart after event
#492 opened by dkweiss31 - 3
Intermediate saved values are sometimes `inf`
#488 opened by dkweiss31 - 7
Coupled SDE System Implementation
#461 opened by aspannaus - 5
Stratonovich/Ito correction
#473 opened by lockwo - 1
Diagonal/Sparse Control Term
#485 opened by SoerenNagel - 2
How to use Jax debug tools inside ODE/SDE/CDE Terms?
#482 opened by qtomcatq - 1
Approaching multi trajectory adaptive stepping
#481 opened by lockwo - 3
Interest in new solvers
#435 opened by lockwo - 3
Neural SDE example does not run
#479 opened by BenjaminJCox - 8
- 4
Use values computed in ODETerms inside of SaveAt
#476 opened by Ricky5389 - 4
- 1
Understand diffrax.diffeqsolve
#467 opened by zhengqigao - 11
Diffrax activating the event on a rejected step
#464 opened by etienney - 1
SpaceTimeTimeLevy Area
#470 opened by lockwo - 1
Adaptive SDE Docs
#471 opened by lockwo - 3
Accelerate ODE solver [What did I miss?]
#466 opened by zhengqigao - 20
statefully evolving an auxiliary variable
#462 opened by jbial - 7
Steady-state event occurs immediately if state with large value constant
#448 opened by johannahaffner - 2
- 3
- 2
Using a Pytree state as a state
#460 opened by ASKabalan - 2
Demos not working
#459 opened by apfelfruchtsaft - 5
- 3
Solving ODE with huge inputs / multi-device
#444 opened by ASKabalan - 3
Warning for solving example ODE from docu
#445 opened by MaAl13 - 1
DAE status
#457 opened by krzysztofrusek - 5
- 1
FutureWarning: unhashable type on jax==0.4.30
#456 opened by gautierronan - 2
- 4
Can't use Equinox inside `term`
#446 opened by pascal-mueller - 2
- 8
Difficulty installing diffrax on apple silicon
#429 opened by cgiovanetti - 2
ImplicitEuler throws RuntimeError with diffeqsolve
#439 opened by JSGandora - 2
Support for uncertainty propagation?
#438 opened by nstarman - 4
How to evaluate derivatives of diffrax-solved, equilibrated functions?
#432 opened by francesco-innocenti - 2
Resolve bear typing warning
#433 opened by lockwo - 2
Possible performance issue with implicit solvers
#428 opened by gautierronan