pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Fast, flexible and easy to use probabilistic modelling in Python.
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Python Toolkit for Causal and Probabilistic Reasoning
Minimalistic large language model 3D-parallelism training
An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries
High-speed Large Language Model Serving for Local Deployment
The hub for EleutherAI's work on interpretability and learning dynamics
3 captures since 2026-05-22