Learn how to design large-scale systems. Prep for the system design interview. Includes Anki flashcards.

Animation engine for explanatory math videos

Rich is a Python library for rich text and beautiful formatting in the terminal.

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Graph Neural Network Library for PyTorch

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable…

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Open source platform for the machine learning lifecycle

End-to-End Object Detection with Transformers

Parallel computing with task scheduling

3D Reconstruction Software

A hyperparameter optimization framework

Statsmodels: statistical modeling and econometrics in Python

Geometric Computer Vision Library for Spatial AI

NumPy aware dynamic Python compiler using LLVM

A collaboration friendly studio for NeRFs

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

Hydra is a framework for elegantly configuring complex applications

Deep universal probabilistic programming with Python and PyTorch

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

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

The open-source tool for building high-quality datasets and computer vision models

A pure-python PDF library capable of splitting, merging, cropping, and transforming the pages of PDF files

A Python implementation of global optimization with gaussian processes.

Uniform Manifold Approximation and Projection

Point cloud diffusion for 3D model synthesis

Fast, correct Python JSON library supporting dataclasses, datetimes, and numpy

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

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