Patrick Kidger's repositories
torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
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.
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"
PatModules.jl
A better import/module system for Julia.
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.
torchdiffeq
Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
mkdocs_include_exclude_files
Modify which files MkDocs includes or excludes.
beartype
Unbearably fast near-real-time runtime type-checking in pure Python.
levanter
Legibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax