🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Flexible and powerful data analysis / manipulation library for Python, providing labeled data structures similar to R data.frame objects, statistical functions, and much more

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

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

A community-maintained Python framework for creating mathematical animations.

Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.

A formatter for Python files

Statsmodels: statistical modeling and econometrics in Python

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

95.47% on CIFAR10 with PyTorch

Progressive Growing of GANs for Improved Quality, Stability, and Variation

A Python library that generates static type annotations by collecting runtime types

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test your data and mo…

Monte Carlo tree search in JAX

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

Differentiable convex optimization layers

Optax is a gradient processing and optimization library for JAX.

"Cyberpunk style" for matplotlib plots

Open Source Pricing & Packaging Infrastructure

Code for the paper "Neural MMO: A Massively Multiagent Game Environment for Training and Evaluating Intelligent Agents"

A simple program which checks Python source files for errors

ARCH models in Python

PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM

Python implementations of the k-modes and k-prototypes clustering algorithms, for clustering categorical data

KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows

Connect to Google Colab using SSH

tree is a library for working with nested data structures

Hardware accelerated, batchable and differentiable optimizers in JAX.

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