Natural Language Processing for the next decade. Tokenization, Part-of-Speech Tagging, Named Entity Recognition, Syntactic & Semantic Dependency Parsing, Document Classification

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

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

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

Use ChatGPT to summarize the arXiv papers. 全流程加速科研，利用chatgpt进行论文全文总结+专业翻译+润色+审稿+审稿回复

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

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Easy-to-use Speech Toolkit including Self-Supervised Learning model, SOTA/Streaming ASR with punctuation, Streaming TTS with text frontend, Speaker Verification System, End-to-End Speech Translatio…

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

An open source implementation of CLIP.

Speech recognition module for Python, supporting several engines and APIs, online and offline.

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

End-to-End Speech Processing Toolkit

A PyTorch implementation of EfficientNet

A faster pytorch implementation of faster r-cnn

BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)

95.47% on CIFAR10 with PyTorch

An Open-Source Framework for Prompt-Learning.

Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet…

Production First and Production Ready End-to-End Speech Recognition Toolkit

An Open-Source Package for Knowledge Embedding (KE)

Official implementation for "Multimodal Chain-of-Thought Reasoning in Language Models" (stay tuned and more will be updated)

OpenMMLab Pre-training Toolbox and Benchmark

Official repository of OFA (ICML 2022). Paper: OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"

The friendly PIL fork

🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch

Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-CAM)

An all-in-one toolkit for computer vision