Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Best Practices on Recommendation Systems

Never use print for debugging again

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Image augmentation for machine learning experiments.

A very simple framework for state-of-the-art Natural Language Processing (NLP)

🐍 Geometric Computer Vision Library for Spatial AI

A PyTorch-based Speech Toolkit

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

Hackable and optimized Transformers building blocks, supporting a composable construction.

TensorFlow-based neural network library

End-to-End Speech Processing Toolkit

Code for the paper Hybrid Spectrogram and Waveform Source Separation

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

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

An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.

PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.

OpenMMLab Pose Estimation Toolbox and Benchmark.

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

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

A data augmentations library for audio, image, text, and video.

Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

Perf monitoring CLI tool for Apple Silicon

Fast Python Collaborative Filtering for Implicit Feedback Datasets

JAX-based neural network library

Deep recommender models using PyTorch.

Neural network visualization toolkit for keras