Stars
🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A collection of design patterns/idioms in Python
💫 Industrial-strength Natural Language Processing (NLP) in Python
An open source implementation of CLIP.
A MNIST-like fashion product database. Benchmark 👇
🐍 Geometric Computer Vision Library for Spatial AI
An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.
Open Source Neural Machine Translation and (Large) Language Models in PyTorch
Benchmarks of approximate nearest neighbor libraries in Python
Probabilistic time series modeling in Python
Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models
Simple examples to introduce PyTorch
A tool for extracting plain text from Wikipedia dumps
ColBERT: state-of-the-art neural search (SIGIR'20, TACL'21, NeurIPS'21, NAACL'22, CIKM'22, ACL'23, EMNLP'23)
A deep learning library for video understanding research.
Speech Recognition using DeepSpeech2.
NCRF++, a Neural Sequence Labeling Toolkit. Easy use to any sequence labeling tasks (e.g. NER, POS, Segmentation). It includes character LSTM/CNN, word LSTM/CNN and softmax/CRF components.
Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP
Metric learning algorithms in Python
Large datasets for conversational AI
curl-like access to AWS resources with AWS Signature Version 4 request signing.
Some custom dataset examples for PyTorch
A PyTorch Implementation of End-to-End Models for Speech-to-Text
RobustBench: a standardized adversarial robustness benchmark [NeurIPS 2021 Benchmarks and Datasets Track]
Code relative to "Reliable evaluation of adversarial robustness with an ensemble of diverse parameter-free attacks"