🤗 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.

🥧 HTTPie CLI — modern, user-friendly command-line HTTP client for the API era. JSON support, colors, sessions, downloads, plugins & more.

OpenMMLab Detection Toolbox and Benchmark

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Code and documentation to train Stanford's Alpaca models, and generate the data.

Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.

Convert Machine Learning Code Between Frameworks

🏄 Scalable embedding, reasoning, ranking for images and sentences with CLIP

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

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.

A Python implementation of global optimization with gaussian processes.

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

A simple and efficient tool to parallelize Pandas operations on all available CPUs

Fast, flexible and easy to use probabilistic modelling in Python.

Accelerated deep learning R&D

Run LLMs with MLX

This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.

PyTorch extensions for fast R&D prototyping and Kaggle farming

Image Test Time Augmentation with PyTorch!

Carnets is a stand-alone Jupyter notebook server and client. Edit your notebooks on the go, even where there is no network.

A library for soundscape synthesis and augmentation

The Catalogs of Sources of the Mobility Database.

Bayesian nonparametric machine learning for Python

Minimal tutorial on packing and unpacking sequences in pytorch

Probabilistic classification in PyTorch/TensorFlow/scikit-learn with Fenchel-Young losses

Python BK-tree data structure to allow fast querying of "close" matches

Keras implementation of the Squeeze Det Object Detection Deep Learning Framework

Pure-python package for map matching

Greedy Gaussian Segmentation