Hacker GitHub

← All repos

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

Mentions per year 8 mentions · busiest year 2022 (5)
20082026
AllStoriesComments NewestTopOldest
  1. 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]
  2. …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?
  3. ▲ 2
    Show HN: Optimistix: Nonlinear Optimisation in Jax+Equinox
    story by packquickly · 2023-10-10 · 1 comment
  4. …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
  5. …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
  6. …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?
  7. …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
  8. …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