duvenaud

jax_fdm

Auto-differentiable and hardware-accelerated force density method

synjax

jump-diffusion

transformer_grammars

Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scale, TACL (2022)

transformer-walkthrough

A walkthrough of transformer architecture code

zuko

Normalizing flows in PyTorch

SDENet-UQ-ESL

SDENet as Uncertainty Quantification Method for EEG Source Localization

sandbox-grounded-qa

A sandbox repo for grounded question answering with Cohere and Google Search

high-order-langevin

High order numerical simulation of the underdamped Langevin diffusion

high-order-splitting

High order numerical simulation of SDEs satisfying a commutativity condition

diffrax

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

bayesoptbook.github.io

Companion webpage for the book "Bayesian Optimization" by Roman Garnett

esh_dynamics

Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling

lietorch

residual-flows

code for "Residual Flows for Invertible Generative Modeling".

time_series_explainability

relearn

Reinforcement learning with Rust

Latent-Neural-Differential-Equations-for-Video-Generation

Official code repository for Latent Neural Differential Equations for Video Generation.

nbody

N body gravity attraction problem solver

strux

yatima-lang-alpha

A programming language for the decentralized web

inference_suboptimality

Replication of Inference Suboptimality in Variational Autoencoders (Cremer, Li, Duvenaud) ATML

graph-neural-pde

Graph Neural PDEs

neural-datalog-through-time

Source code for Neural Datalog Through Time: Informed Temporal Modeling via Logical Specification (ICML 2020).

JAX-BO

HLearn

Homomorphic machine learning

bayeSDE

Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"

mamdp

Code for the paper "How RL Agents Behave When Their Actions Are Modified"

open-data

Free football data from StatsBomb

spaCy

💫 Industrial-strength Natural Language Processing (NLP) in Python