Patrick Kidger (@patrick-kidger)
github

patrick-kidger

Twitter logo Share LinkedIn logo Share Facebook logo Share

Top repositories

1

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Python
1,743
star
2

torchtyping

Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
Python
1,331
star
3

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Python
1,197
star
4

jaxtyping

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

NeuralCDE

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)
Python
576
star
6

torchcde

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

mkposters

Make posters from Markdown files.
Python
317
star
8

sympy2jax

Turn SymPy expressions into trainable JAX expressions.
Python
299
star
9

lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
Python
299
star
10

signatory

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)
C++
246
star
11

optimistix

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

torchcubicspline

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

sympytorch

Turning SymPy expressions into PyTorch modules.
Python
122
star
14

quax

Multiple dispatch over abstract array types in JAX.
Python
88
star
15

FasterNeuralDiffEq

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

Deep-Signature-Transforms

Code for "Deep Signature Transforms" (NeurIPS 2019)
Jupyter Notebook
82
star
17

typst_pyimage

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

generalised_shapelets

Code for "Generalised Interpretable Shapelets for Irregular Time Series"
Jupyter Notebook
51
star
19

PatModules.jl

A better import/module system for Julia.
Julia
17
star
20

action_update_python_project

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

exvoker

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

pytkdocs_tweaks

Some custom tweaks to the results produced by pytkdocs.
Python
5
star
23

Learning-Interpolation

Applying machine learning to help numerically solve the Camassa-Holm equation.
Jupyter Notebook
4
star
24

matching

Round robin matching algorithm.
Python
3
star
25

candle

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

tools

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

pdfscraper

Saves a webpage and all linked pdfs.
Python
3
star
28

ktools

Tools for working with Keras.
Python
2
star
29

loccounter

Counts lines of Python code.
Python
2
star
30

Dissertation

Master's Dissertation: Polynomial Approximation of Holomorphic Functions
2
star
31

py2annotate

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

adventuregame

The very start of a game I was toying with before I got distracted by the PhD...
Python
2
star
33

MPE-CDT-Project

A simple machine learning project for weather observations.
Jupyter Notebook
2
star
34

tfext

Some extra stuff for using with TensorFlow.
Python
2
star
35

mkdocs_include_exclude_files

Modify which files MkDocs includes or excludes.
Python
1
star
36

patrick-kidger

1
star
37

rl-test

Python
1
star