🤗 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)

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

95.47% on CIFAR10 with PyTorch

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.

Open Source Pricing & Packaging Infrastructure

"Cyberpunk style" for matplotlib plots

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.