patrick-kidger

equinox

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

torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.

diffrax

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

jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

torchcde

Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.

mkposters

Make posters from Markdown files.

lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

sympy2jax

Turn SymPy expressions into trainable JAX expressions.

optimistix

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

signatory

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

torchcubicspline

Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, and backpropagation.

sympytorch

Turning SymPy expressions into PyTorch modules.

quax

Multiple dispatch over abstract array types in JAX.

Deep-Signature-Transforms

Code for "Deep Signature Transforms" (NeurIPS 2019)

FasterNeuralDiffEq

Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)

typst_pyimage

Typst extension, adding support for generating figures using inline Python code

generalised_shapelets

Code for "Generalised Interpretable Shapelets for Irregular Time Series"

exvoker

A CLI tool. Extract regexes from stdout (e.g. URLs) and invoke commands on them (e.g. open the webpage).

action_update_python_project

Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI

pytkdocs_tweaks

Some custom tweaks to the results produced by pytkdocs.

eqxvision

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

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

haliax

Named Tensors for Legible Deep Learning in JAX

patrick-kidger

beartype

Unbearably fast near-real-time runtime type-checking in pure Python.

cpython

The Python programming language

JAXFLUIDS

Differentiable Fluid Dynamics Package

levanter

Legibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax

synjax