patrick-kidger
Building a JAX ecosystem: Equinox, Diffrax, Lineax, etc. ---- SciML, numerics, neural ODEs ---- Solo traveller, martial artist, scuba diver!
Cradle.bioZürich
Pinned Repositories
torchsde
Differentiable SDE solvers with GPU support and efficient sensitivity analysis.
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
mkposters
Make posters from Markdown files.
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/
torchcde
Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
patrick-kidger's Repositories
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
patrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
patrick-kidger/NeuralCDE
Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
patrick-kidger/torchcde
Differentiable controlled differential equation solvers for PyTorch with GPU support and memory-efficient adjoint backpropagation.
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
patrick-kidger/mkposters
Make posters from Markdown files.
patrick-kidger/sympy2jax
Turn SymPy expressions into trainable JAX expressions.
patrick-kidger/signatory
Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)
patrick-kidger/torchcubicspline
Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, and backpropagation.
patrick-kidger/sympytorch
Turning SymPy expressions into PyTorch modules.
patrick-kidger/quax
Multiple dispatch over abstract array types in JAX.
patrick-kidger/Deep-Signature-Transforms
Code for "Deep Signature Transforms" (NeurIPS 2019)
patrick-kidger/FasterNeuralDiffEq
Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
patrick-kidger/typst_pyimage
Typst extension, adding support for generating figures using inline Python code
patrick-kidger/generalised_shapelets
Code for "Generalised Interpretable Shapelets for Irregular Time Series"
patrick-kidger/action_update_python_project
Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI
patrick-kidger/exvoker
A CLI tool. Extract regexes from stdout (e.g. URLs) and invoke commands on them (e.g. open the webpage).
patrick-kidger/pytkdocs_tweaks
Some custom tweaks to the results produced by pytkdocs.
patrick-kidger/eqxvision
A Python package for computer vision models in the Equinox ecosystem.
patrick-kidger/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
patrick-kidger/haliax
Named Tensors for Legible Deep Learning in JAX
patrick-kidger/patrick-kidger
patrick-kidger/beartype
Unbearably fast near-real-time runtime type-checking in pure Python.
patrick-kidger/cpython
The Python programming language
patrick-kidger/JAXFLUIDS
Differentiable Fluid Dynamics Package
patrick-kidger/levanter
Legibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax
patrick-kidger/synjax