Owner · patrick-kidger

patrick-kidger

patrick-kidger's tracked open-source repos, sorted by stars.

7 repos

#1
equinox ↗@patrick-kidgerPython
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
★ 2,963+2Star change over the last 7 days
#2
diffrax ↗@patrick-kidgerPython
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
★ 2,091+0Star change over the last 7 days
#3
jaxtyping ↗@patrick-kidgerPython
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
★ 1,863+1Star change over the last 7 days
#4
torchtyping ↗@patrick-kidgerPython
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
★ 1,482+0Star change over the last 7 days
#5
tinyio ↗@patrick-kidgerPython
Ever used asyncio and wished you hadn't? A tiny (~400 lines) event loop for Python.
★ 739+1Star change over the last 7 days
#6
optimistix ↗@patrick-kidgerPython
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
★ 618+2Star change over the last 7 days
#7
lineax ↗@patrick-kidgerPython
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
★ 578+1Star change over the last 7 days