Acknowledgements

Diffrax is inspired by previous differential equation libraries written for other frameworks. In particular:

• torchdiffeq
• torchsde
• torchcde
• DifferentialEquations.jl

Thanks to the JAX team for the framework that makes this possible.