patrick-kidger/diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Language: Python - Size: 5.42 MB - Last synced: 2 days ago - Pushed: 2 days ago - Stars: 1,253 - Forks: 111

patrick-kidger/equinox

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

Language: Python - Size: 17.4 MB - Last synced: 3 days ago - Pushed: 3 days ago - Stars: 1,835 - Forks: 127

patrick-kidger/optimistix

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

Language: Python - Size: 967 KB - Last synced: 3 days ago - Pushed: 4 days ago - Stars: 248 - Forks: 10

patrick-kidger/lineax

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

Language: Python - Size: 196 KB - Last synced: about 17 hours ago - Pushed: 7 days ago - Stars: 315 - Forks: 18

patrick-kidger/typst_pyimage

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

Language: Python - Size: 68.4 KB - Last synced: 7 days ago - Pushed: 7 days ago - Stars: 67 - Forks: 1

patrick-kidger/jaxtyping

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

Language: Python - Size: 278 KB - Last synced: 9 days ago - Pushed: 9 days ago - Stars: 956 - Forks: 43

patrick-kidger/Deep-Signature-Transforms

Code for "Deep Signature Transforms" (NeurIPS 2019)

Language: Jupyter Notebook - Size: 560 KB - Last synced: 12 days ago - Pushed: 12 days ago - Stars: 85 - Forks: 20

patrick-kidger/PatModules.jl

A better import/module system for Julia.

Language: Julia - Size: 56.6 KB - Last synced: 7 days ago - Pushed: over 3 years ago - Stars: 18 - Forks: 1

patrick-kidger/torchtyping

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

Language: Python - Size: 101 KB - Last synced: 13 days ago - Pushed: 11 months ago - Stars: 1,337 - Forks: 31

patrick-kidger/torchsde Fork of google-research/torchsde

Differentiable SDE solvers with GPU support and efficient sensitivity analysis.

Language: Python - Size: 3.56 MB - Last synced: 15 days ago - Pushed: over 3 years ago - Stars: 3 - Forks: 2

patrick-kidger/torchdiffeq Fork of rtqichen/torchdiffeq

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

Size: 8.11 MB - Last synced: 15 days ago - Pushed: about 3 years ago - Stars: 3 - Forks: 0

patrick-kidger/tools 📦

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

Language: Python - Size: 96.7 KB - Last synced: 15 days ago - Pushed: almost 5 years ago - Stars: 3 - Forks: 2

patrick-kidger/tfext 📦

Some extra stuff for using with TensorFlow.

Language: Python - Size: 132 KB - Last synced: 15 days ago - Pushed: over 5 years ago - Stars: 2 - Forks: 1

patrick-kidger/sacred Fork of IDSIA/sacred

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

Size: 3.48 MB - Last synced: 15 days ago - Pushed: about 3 years ago - Stars: 2 - Forks: 0

patrick-kidger/pdfscraper

Saves a webpage and all linked pdfs.

Language: Python - Size: 8.79 KB - Last synced: 15 days ago - Pushed: over 5 years ago - Stars: 3 - Forks: 1

patrick-kidger/MPE-CDT-Project 📦

A simple machine learning project for weather observations.

Language: Jupyter Notebook - Size: 22.2 MB - Last synced: 15 days ago - Pushed: almost 6 years ago - Stars: 2 - Forks: 1

patrick-kidger/loccounter 📦

Counts lines of Python code.

Language: Python - Size: 13.7 KB - Last synced: 15 days ago - Pushed: almost 5 years ago - Stars: 2 - Forks: 1

patrick-kidger/ktools 📦

Tools for working with Keras.

Language: Python - Size: 87.9 KB - Last synced: 15 days ago - Pushed: about 5 years ago - Stars: 2 - Forks: 1

patrick-kidger/JAXFLUIDS Fork of tumaer/JAXFLUIDS

Differentiable Fluid Dynamics Package

Size: 8.89 MB - Last synced: 15 days ago - Pushed: over 1 year ago - Stars: 0 - Forks: 0

patrick-kidger/jax Fork of google/jax

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

Language: Python - Size: 68.1 MB - Last synced: 15 days ago - Pushed: 5 months ago - Stars: 2 - Forks: 0

patrick-kidger/FromFile.jl Fork of Roger-luo/FromFile.jl

Language: Julia - Size: 25.4 KB - Last synced: 15 days ago - Pushed: over 3 years ago - Stars: 1 - Forks: 0

patrick-kidger/Dissertation 📦

Master's Dissertation: Polynomial Approximation of Holomorphic Functions

Size: 399 KB - Last synced: 15 days ago - Pushed: almost 6 years ago - Stars: 2 - Forks: 1

patrick-kidger/cpython Fork of python/cpython

The Python programming language

Size: 532 MB - Last synced: 15 days ago - Pushed: about 1 year ago - Stars: 0 - Forks: 0

patrick-kidger/candle 📦

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

Language: Python - Size: 13.7 KB - Last synced: 15 days ago - Pushed: almost 5 years ago - Stars: 3 - Forks: 1

patrick-kidger/adabound Fork of titu1994/keras-adabound

Keras implementation of AdaBound

Language: Python - Size: 3.5 MB - Last synced: 15 days ago - Pushed: about 5 years ago - Stars: 2 - Forks: 0

patrick-kidger/adventuregame

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

Language: Python - Size: 456 KB - Last synced: 15 days ago - Pushed: almost 6 years ago - Stars: 2 - Forks: 1

patrick-kidger/NeuralCDE

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

Language: Python - Size: 615 KB - Last synced: 15 days ago - Pushed: over 1 year ago - Stars: 583 - Forks: 68

patrick-kidger/signatory

Differentiable computations of the signature and logsignature transforms, on both CPU and GPU. (ICLR 2021)

Language: C++ - Size: 1.4 MB - Last synced: 13 days ago - Pushed: 4 months ago - Stars: 250 - Forks: 30

patrick-kidger/sympy2jax

Turn SymPy expressions into trainable JAX expressions.

Language: Python - Size: 26.4 KB - Last synced: 13 days ago - Pushed: 26 days ago - Stars: 308 - Forks: 12

patrick-kidger/mkposters

Make posters from Markdown files.

Language: Python - Size: 6.13 MB - Last synced: 13 days ago - Pushed: about 1 year ago - Stars: 322 - Forks: 21

patrick-kidger/quax

Multiple dispatch over abstract array types in JAX.

Language: Python - Size: 145 KB - Last synced: 13 days ago - Pushed: 26 days ago - Stars: 91 - Forks: 2

patrick-kidger/torchcubicspline

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

Language: Python - Size: 35.2 KB - Last synced: 15 days ago - Pushed: over 1 year ago - Stars: 185 - Forks: 17

patrick-kidger/torchcde

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

Language: Python - Size: 247 KB - Last synced: 13 days ago - Pushed: 10 months ago - Stars: 393 - Forks: 45

patrick-kidger/exvoker

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

Language: Rust - Size: 452 KB - Last synced: 15 days ago - Pushed: about 1 year ago - Stars: 9 - Forks: 0

patrick-kidger/generalised_shapelets

Code for "Generalised Interpretable Shapelets for Irregular Time Series"

Language: Jupyter Notebook - Size: 42.5 MB - Last synced: 15 days ago - Pushed: 10 months ago - Stars: 52 - Forks: 10

patrick-kidger/sympytorch

Turning SymPy expressions into PyTorch modules.

Language: Python - Size: 59.6 KB - Last synced: 15 days ago - Pushed: 10 months ago - Stars: 124 - Forks: 20

patrick-kidger/patrick-kidger

Size: 6.84 KB - Last synced: 15 days ago - Pushed: about 2 months ago - Stars: 1 - Forks: 0

patrick-kidger/matching 📦

Round robin matching algorithm.

Language: Python - Size: 8.79 KB - Last synced: 15 days ago - Pushed: over 5 years ago - Stars: 3 - Forks: 1

patrick-kidger/FasterNeuralDiffEq

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

Language: Python - Size: 635 KB - Last synced: 15 days ago - Pushed: over 1 year ago - Stars: 84 - Forks: 9

patrick-kidger/action_update_python_project

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

Size: 27.3 KB - Last synced: 15 days ago - Pushed: 7 months ago - Stars: 8 - Forks: 2

patrick-kidger/synjax Fork of google-deepmind/synjax

Size: 111 KB - Last synced: 15 days ago - Pushed: 9 months ago - Stars: 0 - Forks: 0

patrick-kidger/beartype Fork of beartype/beartype

Unbearably fast near-real-time runtime type-checking in pure Python.

Size: 7.51 MB - Last synced: 15 days ago - Pushed: 10 months ago - Stars: 0 - Forks: 0

patrick-kidger/eqxvision Fork of paganpasta/eqxvision

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

Size: 1.77 MB - Last synced: 15 days ago - Pushed: over 1 year ago - Stars: 3 - Forks: 0

patrick-kidger/rl-test

Language: Python - Size: 5.86 KB - Last synced: 15 days ago - Pushed: over 3 years ago - Stars: 1 - Forks: 0

patrick-kidger/haliax Fork of stanford-crfm/haliax

Named Tensors for Legible Deep Learning in JAX

Size: 163 KB - Last synced: 15 days ago - Pushed: 10 months ago - Stars: 1 - Forks: 0

patrick-kidger/levanter Fork of stanford-crfm/levanter

Legibile, Scalable, Reproducible Foundation Models with Named Tensors and Jax

Size: 5.15 MB - Last synced: 15 days ago - Pushed: 10 months ago - Stars: 0 - Forks: 0

patrick-kidger/Learning-Interpolation 📦

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

Language: Jupyter Notebook - Size: 1020 KB - Last synced: 15 days ago - Pushed: over 5 years ago - Stars: 4 - Forks: 1

patrick-kidger/pytkdocs_tweaks

Some custom tweaks to the results produced by pytkdocs.

Language: Python - Size: 23.4 KB - Last synced: 1 day ago - Pushed: 5 months ago - Stars: 5 - Forks: 2

patrick-kidger/py2annotate

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

Language: Python - Size: 82 KB - Last synced: 14 days ago - Pushed: over 4 years ago - Stars: 2 - Forks: 1

patrick-kidger/mkdocs_include_exclude_files

Modify which files MkDocs includes or excludes.

Language: Python - Size: 6.84 KB - Last synced: 3 days ago - Pushed: over 2 years ago - Stars: 1 - Forks: 0