patrick-kidger/diffrax
View on GitHub ↗ ★ 2,122 Python
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
8 mentions on Hacker News by 3 people: 1 story, 7 comments · first 2022-04-14 · latest 2025-01-11
20082026
-
Did you use https://github.com/patrick-kidger/diffrax ? JAX Talk: Diffrax https://www.youtube.com/watch?v=Jy5Jw8hNiAQ
comment by sitkack · 2025-01-11 · in An Introduction to Neural Ordinary Differential Equations [pdf] -
…a bit of a foundation for a JAX sciML ecosystem. [1] https://github.com/patrick-kidger/equinox [2] https://github.com/patrick-kidger/diffrax
comment by patrickkidger · 2023-12-04 · in Ask HN: What side projects landed you a job? -
…reduced Python-level overhead.) E.g for numerical ODE simulation, I've found that Diffrax ( https://github.com/patrick-kidger/diffrax ) is ~100 times faster than torchdiffeq on the forward pass. The backward pass is much closer, and for…
comment by patrickkidger · 2022-12-02 · in PyTorch 2.0 -
…here: shameless self-promotion for Equinox and Diffrax: https://github.com/patrick-kidger/equinox https://github.com/patrick-kidger/diffrax Which are neural network and differential equation libraries for JAX. [Obligatory…
comment by patrickkidger · 2022-10-26 · in Python 3.11 is faster than 3.8 -
…doing all my work in JAX. (E.g. see Equinox or Diffrax: https://github.com/patrick-kidger/equinox , https://github.com/patrick-kidger/diffrax ). A lot of modern work is now based around hybridising such techniques with neural networks. I'd…
comment by patrickkidger · 2022-10-15 · in Ask HN: What's your favorite programmer niche? -
…of it, but if you want differential-equations-in-JAX then allow me to quickly advertise Diffrax: https://github.com/patrick-kidger/diffrax (of which I am the author, disclaimer).
comment by patrickkidger · 2022-05-16 · in Correctness and composability bugs in the Julia ecosystem -
…overhead whilst solving the above issue very neatly. For example this is precisely what Diffrax ( https://github.com/patrick-kidger/diffrax ) does under `jax.vmap`. I've not dug into what Julia does here; is this not already done when…
comment by patrickkidger · 2022-04-14 · in Doing small network scientific machine learning in Julia faster than PyTorch