Patrick Kidger

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

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/

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

tinyio. Ever used asyncio and wished you hadn't? A tiny (~400 lines) event loop for Python.

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

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

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

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

mkposters. Make posters from Markdown files.

sympy2jax. Turn SymPy expressions into trainable JAX expressions.

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.

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

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)

patdb. A snappy + easy + pretty TUI debugger for Python.

generalised_shapelets. Code for "Generalised Interpretable Shapelets for Irregular Time Series"

wadler_lindig. A Wadler–Lindig pretty printer for Python

esm2quinox. An implementation of ESM2 in Equinox+JAX

typsy. Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.

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

PatModules.jl. A better import/module system for Julia.

mkdocs_ipynb. MkDocs for ipynb files

hippogriffe. Tweaks for `mkdocstrings[python]`

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

pytkdocs_tweaks. Some custom tweaks to the results produced by pytkdocs.

Learning-Interpolation. Applying machine learning to help numerically solve the Camassa-Holm equation.

torchsde. Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

torchdiffeq. Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

pdfscraper. Saves a webpage and all linked pdfs.

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

tools. Helpful abstract tools (functions, classes, ... ) for coding in Python.

matching. Round robin matching algorithm.

candle. Simple PyTorch helpers. (I think we've probably all written one of these for ourselves!)

mkdocs_include_exclude_files. Modify which files MkDocs includes or excludes.

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

adabound. Keras implementation of AdaBound

sacred. Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

py2annotate. An extension to Sphinx autodoc to augment Sphinx documentation with type annotations, when using Python 2 style type annotations.

Dissertation. Master's Dissertation: Polynomial Approximation of Holomorphic Functions

tfext. Some extra stuff for using with TensorFlow.

rl-test. Python

haliax. Named Tensors for Legible Deep Learning in JAX

flytekit. Extensible Python SDK for developing Flyte tasks and workflows. Simple to get started and learn and highly extensible.

cpython. The Python programming language