Hacker GitHub

← All repos

patrick-kidger/equinox

View on GitHub ↗ ★ 2,997 Python

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

15 mentions on Hacker News by 5 people: 3 stories, 12 comments · first 2022-10-15 · latest 2025-01-11

Mentions per year 15 mentions · busiest year 2023 (11)
20082026
AllStoriesComments NewestTopOldest
  1. …`vmap`-able differential equation solving is really cool. [1]: https://dynamicfieldtheory.org/ [2]: https://github.com/patrick-kidger/equinox
    comment by yberreby · 2025-01-11 · in An Introduction to Neural Ordinary Differential Equations [pdf]
  2. There's quite a few other libraries associated with Equinox in the JAX ecosystem: https://github.com/patrick-kidger/equinox?tab=readme-ov-file... I've enjoyed using Equinox and Diffrax for performing ODE simulations. To my knowledge the…
    comment by ubj · 2024-09-12 · in Jax and Equinox: What are they and why should I bother?
  3. …other JAX libraries and we've turned this into 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?
  4. If you like PyTorch then you might like Equinox, by the way. ( https://github.com/patrick-kidger/equinox ; 1.4k GitHub stars now!) Basically designed to offer PyTorch-like syntax for working with JAX. The latter is excellent for the…
    comment by patrickkidger · 2023-09-29 · in JAX – NumPy on the CPU, GPU, and TPU
  5. You can do JAX->TF->ONNX I believe: https://github.com/patrick-kidger/equinox/blob/main/equinox/...
    comment by patrickkidger · 2023-09-29 · in JAX – NumPy on the CPU, GPU, and TPU
  6. …just don't expect to do in-place mutation. (I.e. OO-style programming.) You might like Equinox ( https://github.com/patrick-kidger/equinox ; 1.4k GitHub stars) which deliberately offers a very PyTorch-like feel for JAX. Regarding speed, I…
    comment by patrickkidger · 2023-09-29 · in JAX – NumPy on the CPU, GPU, and TPU
  7. …that's JAX or something else. My experience might be biased though: shameless advert for Equinox ( https://github.com/patrick-kidger/equinox , 1.4k GitHub stars), which is now the foundation of quite a lot of SciComp in JAX. (Both internal…
    comment by patrickkidger · 2023-09-29 · in JAX – NumPy on the CPU, GPU, and TPU
  8. …that you can backprop through! The secret sauce is to use a treeverse-style checkpointing scheme. https://github.com/patrick-kidger/equinox/blob/f95a8ba13fb35...
    comment by patrickkidger · 2023-05-22 · in Pytrees
  9. Shameless advert -- Equinox is a neural network library for JAX based entirely around pytrees: https://github.com/patrick-kidger/equinox (Now on 1.1k stars so it's achieved some popularity!) This makes model-building elegant (IMO), without…
    comment by patrickkidger · 2023-05-22 · in Pytrees
  10. …own thing? (E.g. `haiku.grad`.) If so, then allow me to make my usual advert here for Equinox: https://github.com/patrick-kidger/equinox This actually works with JAX's native transformations. (There's no `equinox.vmap` for example.) On…
    comment by patrickkidger · 2023-02-25 · in Training Deep Networks with Data Parallelism in Jax
  11. …100% of my work in these days. Whilst I'm 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…
    comment by patrickkidger · 2022-10-26 · in Python 3.11 is faster than 3.8
  12. …LQP, molecular dynamics, ... These days I'm 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…
    comment by patrickkidger · 2022-10-15 · in Ask HN: What's your favorite programmer niche?