matt-graham

Matt Graham's starred repositories

jax

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

Language:PythonApache-2.029205 329 5333

torchdiffeq

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

Language:PythonMIT5368 127 214

numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Language:PythonApache-2.02098 51 747

autodidact

A pedagogical implementation of Autograd

Language:Jupyter NotebookMIT930 17 4

blackjax

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

Language:PythonApache-2.0758 14 301

Cassette.jl

Overdub Your Julia Code

Language:JuliaNOASSERTION371 24 112

extending-jax

Extending JAX with custom C++ and CUDA code

Language:PythonMIT363 10 6

cola

Compositional Linear Algebra

Language:PythonApache-2.0325 4 41

zeus

⚡️ zeus: Lightning Fast MCMC ⚡️

Language:PythonGPL-3.0220 9 26

minimc

Just a little MCMC

Language:PythonMIT207 11 3

rise

RISE: "Live" Reveal.js JupyterLab Slideshow Extension

Language:TypeScriptBSD-3-Clause159 10 42

ad-handbook

Automatic Differentiation Handbook

Language:C++NOASSERTION143 11 5

s2fft

Differentiable and accelerated spherical transforms with JAX

Language:PythonMIT120 9 109

pyvbmc

PyVBMC: Variational Bayesian Monte Carlo algorithm for posterior and model inference in Python

Language:PythonBSD-3-Clause110 5 12

sunode

Solve ODEs fast, with support for PyMC

Language:Jupyter NotebookMIT103 21 19

ADBench

Benchmarking various AD tools.

Language:C++MIT102 16 57

Pigeons.jl

Distributed and parallel sampling from intractable distributions

Language:JuliaAGPL-3.073 8 96

SequentialMonteCarlo.jl

A light interface to serial and multi-threaded Sequential Monte Carlo

Language:JuliaNOASSERTION30 5 1

IPUToolkit.jl

IPU programming in Julia

Language:JuliaMIT30 6 23

python-tooling

Python package template for new research software projects

Language:PythonMIT28 17 132

fenics-pymc3

Differentiable interface to FEniCS for PyMC3

Language:PythonMIT20 4 4

composite-actions

Templates for commonly used GitHub actions steps

Language:Python12 190

GPfY

Gaussian processes with spherical harmonic features in JAX

Language:PythonApache-2.011 50

stat_comp_benchmarks

Benchmark Models for Evaluating Algorithm Accuracy

Language:RBSD-3-Clause10 26 1

TLOmodel

Epidemiology modelling framework for the Thanzi la Onse project

Language:PythonMIT10 5 522

demcmc

Differential Emission Measure estimation using MCMC methods

Language:PythonBSD-3-Clause8 2 32

stan-model-server

lightweight server interface to Stan model methods

Language:C++BSD-3-Clause3 7 3

USS

Data on Universities Superannuation Scheme

Language:Jupyter Notebook3 20

dxh

Collection of helper functions for working with DOLFINx Python interface

Language:PythonMIT3 4 8

3d_very_deep_vae

PyTorch implementations of variational autoencoders for 3D images

Language:PythonGPL-3.02 3 20