🤗 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

Robust Speech Recognition via Large-Scale Weak Supervision

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

🎨 Diagram as Code for prototyping cloud system architectures

Making large AI models cheaper, faster and more accessible

The uncompromising Python code formatter

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamically to do so.

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.

The lean application framework for Python. Build sophisticated user interfaces with a simple Python API. Run your apps in the terminal and a web browser.

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.

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

AI orchestration framework to build customizable, production-ready LLM applications. Connect components (models, vector DBs, file converters) to pipelines or agents that can interact with your data…

Turn (almost) any Python command line program into a full GUI application with one line

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

Fast and memory-efficient exact attention

PyScript is an open source platform for Python in the browser. Try PyScript: https://pyscript.com Examples: https://tinyurl.com/pyscript-examples Community: https://discord.gg/HxvBtukrg2

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

🦉 Data Versioning and ML Experiments

DALL·E Mini - Generate images from a text prompt

Memray is a memory profiler for Python

Convert Machine Learning Code Between Frameworks

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

Python package built to ease deep learning on graph, on top of existing DL frameworks.

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RN…

Open source code for AlphaFold 2.