Top 4 Python equinox Projects

equinox

31 1,809 9.2 Python

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

Project mention: Ask HN: What side projects landed you a job? | news.ycombinator.com | 2023-12-03

I wrote a JAX-based neural network library (Equinox [1]) and numerical differential equation solving library (Diffrax [2]).
At the time I was just exploring some new research ideas in numerics -- and frankly, procrastinating from writing up my PhD thesis!
But then one of the teams at Google starting using them, so they offered me a job to keep developing them for their needs. Plus I'd get to work in biotech, which was a big interest of mine. This was a clear dream job offer, so I accepted.
Since then both have grown steadily in popularity (~2.6k GitHub stars) and now see pretty widespread use! I've since started writing several other JAX libraries and we now have a bit of an ecosystem going.
[1] https://github.com/patrick-kidger/equinox

diffrax

21 1,230 8.3 Python

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Project mention: Ask HN: What side projects landed you a job? | news.ycombinator.com | 2023-12-03

WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
optimistix

2 237 8.5 Python

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Project mention: [P] Optimistix, nonlinear optimisation in JAX+Equinox! | /r/MachineLearning | 2023-10-14

Hi everyone! I wanted to advertise my new JAX optimisation library Optimistix!

eqxvision

3 95 2.3 Python

A Python package of computer vision models for the Equinox ecosystem.

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).