Patrick Kidger

Core Contributor to JAX ecosystem

Authored Projects (6)

Starred by

Roy Frostig

Roy Frostig(Coauthor of JAX; Research Scientist at Google DeepMind)

Patrick von Platen

Patrick von Platen(Author of Hugging Face Diffusers; Research Engineer at Mistral)

Phil Wang

Phil Wang(Prolific Research Paper Implementer)

diffrax by patrick-kidger

JAX library for numerical differential equation solvers

Solves ODEs, SDEs, and CDEs using a unified approach.
Offers many solvers: Tsit5, Dopri8, symplectic, and implicit.
Supports PyTree state, dense solutions, and adjoint backpropagation.
Compatible with neural differential equations.

Created 4 years ago

Updated 6 days ago

Starred by

Phil Wang

Phil Wang(Prolific Research Paper Implementer).

optimistix by patrick-kidger

JAX library for nonlinear solvers (root finding, least squares, etc.)

Interoperable solvers: autoconvert root find problems to least squares.
Modular optimizers: BFGS quadratic bowl with dogleg descent path.
Uses PyTrees as state; fast compilation and runtimes.
Interoperable with Optax; supports autodiff, autoparallelism, GPU/TPU.

Created 1 year ago

Updated 2 weeks ago

Starred by

Georgios Konstantopoulos

Georgios Konstantopoulos(CTO, General Partner at Paradigm).

lineax by patrick-kidger

JAX library for linear solves and least squares

Solves Ax=b, even for ill-posed/rectangular A, using PyTree-valued matrices/vectors.
Employs general linear operators for Jacobians and transposes.
Offers efficient, stable gradients via QR solvers; supports structured matrices.
Leverages JAX for autodiff, autoparallelism, and GPU/TPU support.

Created 2 years ago

Updated 3 weeks ago

Starred by

Matthew Johnson

Matthew Johnson(Coauthor of JAX; Research Scientist at Google Brain),

Jeff Hammerbacher

Jeff Hammerbacher(Cofounder of Cloudera),

Jonathan Ragan-Kelley

Jonathan Ragan-Kelley(Professor at MIT),

Hanlin Tang

Hanlin Tang(CTO Neural Networks at Databricks; Cofounder of MosaicML), and

3 more.

equinox by patrick-kidger

JAX library for neural networks and scientific computing

PyTorch-like syntax for defining models as PyTrees.
Filtered APIs for transformations, PyTree manipulation routines.
Compatible with JAX ecosystem; supports runtime errors.

Created 4 years ago

Updated 3 weeks ago

Starred by

George Hotz

George Hotz(Author of tinygrad; Founder of the tiny corp, comma.ai),

Patrick von Platen

Patrick von Platen(Author of Hugging Face Diffusers; Research Engineer at Mistral),

Eric Zhang

Eric Zhang(Founding Engineer at Modal),

Phil Wang

Phil Wang(Prolific Research Paper Implementer), and

3 more.

jaxtyping by patrick-kidger

Typing library for array shapes/dtypes

Annotates JAX, PyTorch, NumPy, MLX, TensorFlow arrays.
Supports PyTree typing for nested data structures.
Integrates with runtime type-checking packages.

Created 3 years ago

Updated 4 months ago

sympy2jax by patrick-kidger

JAX module for symbolic computation

Converts SymPy expressions into trainable Equinox modules.
Enables gradient-based optimization of symbolic expressions.
Supports custom SymPy-to-JAX function translations.

Created 3 years ago

Updated 4 months ago

Starred Projects (26)

ubicloud by ubicloud

Open-source IaaS alternative to AWS

gitbuda:

dguido:

artman:

hsbt:

Created 2 years ago

Updated 15 hours ago

aim by aimhubio

Experiment tracker for AI model training runs

amin3141:

transitive-bullshit:

yang-song:

stas00:

Created 6 years ago

Updated 1 day ago

einx by fferflo

Tensor operation library using Einstein-inspired notation

awni:

jiamings:

froystig:

hammer:

Created 2 years ago

Updated 5 months ago

ThunderKittens by HazyResearch

CUDA kernel framework for fast deep learning primitives

karpathy:

gakonst:

luiscape:

albertfgu:

Created 1 year ago

Updated 2 days ago

dspy by stanfordnlp

Framework for programming language models, not prompting

tobi:

mateiz:

vincentweisser:

pgarbacki:

Created 2 years ago

Updated 15 hours ago

optimistix by patrick-kidger

JAX library for nonlinear solvers (root finding, least squares, etc.)

lucidrains:

Created 1 year ago

Updated 2 weeks ago

curated-transformers by explosion

PyTorch library for composable Transformer models

simonw:

jph00:

rromb:

millionintegrals:

Created 3 years ago

Updated 1 year ago

levanter by marin-community

Framework for training foundation models with JAX

pcmoritz:

Jiayi-Pan:

thomwolf:

chiphuyen:

Created 3 years ago

Updated 18 hours ago

lineax by patrick-kidger

JAX library for linear solves and least squares

gakonst:

Created 2 years ago

Updated 3 weeks ago

chatblade by npiv

CLI tool for interacting with OpenAI's ChatGPT

AntonOsika:

gakonst:

Created 2 years ago

Updated 11 months ago

PhySO by WassimTenachi

Symbolic optimization package for physics

Created 2 years ago

Updated 1 week ago

mltype by jankrepl

CLI tool for improving typing speed, especially for programmers

Created 5 years ago

Updated 2 years ago

jaxtyping by patrick-kidger

Typing library for array shapes/dtypes

geohot:

patrickvonplaten:

ekzhang:

lucidrains:

Created 3 years ago

Updated 4 months ago

sympy2jax by patrick-kidger

JAX module for symbolic computation

Created 3 years ago

Updated 4 months ago

composer by mosaicml

DL framework for training at scale, optimized for large-scale clusters

aravindsrinivas:

eiso:

gakonst:

jn2clark:

Created 3 years ago

Updated 2 weeks ago

equinox by patrick-kidger

JAX library for neural networks and scientific computing

mattjj:

hammer:

jrk:

hanlint:

Created 4 years ago

Updated 3 weeks ago

Awesome-Diffusion-Models by diff-usion

Diffusion models resource collection

omarsar:

taranjeet:

osanseviero:

eugeneyan:

Created 4 years ago

Updated 1 year ago

diffrax by patrick-kidger

JAX library for numerical differential equation solvers

froystig:

patrickvonplaten:

lucidrains:

Created 4 years ago

Updated 6 days ago

scalene by plasma-umass

Python profiler with AI-powered optimization proposals

luiscape:

xiezhq-hermann:

zhuohan123:

Ying1123:

Created 5 years ago

Updated 1 week ago

x-transformers by lucidrains

Transformer library with extensive experimental features

chiphuyen:

hammer:

chenlin9:

Jiayi-Pan:

Created 4 years ago

Updated 5 days ago

WeightWatcher by CalculatedContent

DNN analyzer for predicting model accuracy without training/test data

hammer:

lantiga:

Created 6 years ago

Updated 3 months ago

glow by openai

Generative flow research paper code

lilianweng:

tgaddair:

Edward-Sun:

chenlin9:

Created 7 years ago

Updated 1 year ago

gym-anytrading by AminHP

Gym environments for RL-based trading algorithms (FOREX, Stocks)

pathak22:

Created 6 years ago

Updated 1 year ago

umap by lmcinnes

Dimension reduction technique for visualization and general non-linear reduction

transitive-bullshit:

lewtun:

ankane:

domoritz:

Created 8 years ago

Updated 4 days ago

awesome-neural-ode by Zymrael

Resource list for neural differential equations and scientific ML

ablattmann:

rromb:

lantiga:

Created 6 years ago

Updated 1 year ago

tqdm by tqdm

Progress bar for Python and CLI tools

bcherny:

geohot:

hugs:

parano:

Created 10 years ago

Updated 3 months ago

Feedback? Help us improve.