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)
sympy2jax
Turn SymPy expressions into trainable JAX expressions.
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/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
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/mkposters
Make posters from Markdown files.
patrick-kidger/sympy2jax
Turn SymPy expressions into trainable JAX expressions.
patrick-kidger/optimistix
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
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/exvoker
A CLI tool. Extract regexes from stdout (e.g. URLs) and invoke commands on them (e.g. open the webpage).
patrick-kidger/action_update_python_project
Github Action to: Check version / Test / git tag / GitHub Release / Deploy to PyPI
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