Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

An open-source NLP research library, built on PyTorch.

Super Resolution for images using deep learning.

A curated list of awesome mathematics resources

Style transfer, deep learning, feature transform

Statsmodels: statistical modeling and econometrics in Python

Cleanlab's open-source library is the standard data-centric AI package for data quality and machine learning with messy, real-world data and labels.

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

Minimal and clean examples of machine learning algorithms implementations

🐍 Geometric Computer Vision Library for Spatial AI

instant coding answers via the command line

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

YOLOv3 in PyTorch > ONNX > CoreML > TFLite

A little word cloud generator in Python

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Refine high-quality datasets and visual AI models

Hydra is a framework for elegantly configuring complex applications

TensorFlow-based neural network library

End-to-End Speech Processing Toolkit

A PyTorch implementation of the Transformer model in "Attention is All You Need".

Code for the paper Hybrid Spectrogram and Waveform Source Separation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

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

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image …

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

Deep universal probabilistic programming with Python and PyTorch

PyTorch tutorials.

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

Web mining module for Python, with tools for scraping, natural language processing, machine learning, network analysis and visualization.