Really Fast End-to-End Jax RL Implementations

Get up and running with Kimi-K2.6, GLM-5.2, MiniMax, DeepSeek, gpt-oss, Qwen, Gemma and other models.

Index your Gmail account to a SQLite DB and play with the data.

A native Julia code for lattice QCD with dynamical fermions in 4 dimension.

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

Boltzmann Generators and Normalizing Flows in PyTorch

PyTorch implementation of normalizing flow models

Normalizing flows in PyTorch

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

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

A simulation-based Inference (SBI) library designed to perform analysis on stellar streams in the Milky Way

High-Performance Symbolic Regression in Python and Julia

Error propagation and statistical analysis for Monte Carlo simulations in lattice QCD and statistical mechanics using autograd.

Bayesian Modeling and Probabilistic Programming in Python

jax library for E3 Equivariant Neural Networks

[NeurIPS 2021] Equivariant Manifold Flows (https://arxiv.org/abs/2107.08596)

Paper list for equivariant neural network

Code for the paper "Learning Differential Equations that are Easy to Solve"

A python library for studying Calabi-Yau metrics

Neural Monte Carlo Renormalization Group

A system for scientific simulation-based inference at scale.

Flax is a neural network library for JAX that is designed for flexibility.

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

IMAP (version 4rev1) implementation for dart