🤗 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

Animation engine for explanatory math videos

Models and examples built with TensorFlow

Deep Learning for humans

Build and share delightful machine learning apps, all in Python. 🌟 Star to support our work!

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

A toolkit for developing and comparing reinforcement learning algorithms.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

⚡ A Fast, Extensible Progress Bar for Python and CLI

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source …

The open source developer platform to build AI agents and models with confidence. Enhance your AI applications with end-to-end tracking, observability, and evaluations, all in one integrated platform.

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

Open standard for machine learning interoperability

Magenta: Music and Art Generation with Machine Intelligence

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

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

Parallel computing with task scheduling

Redis Python client

1 Line of code data quality profiling & exploratory data analysis for Pandas and Spark DataFrames.

Build, Manage and Deploy AI/ML Systems

A python library for user-friendly forecasting and anomaly detection on time series.

Deep universal probabilistic programming with Python and PyTorch

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

Efficiently computes derivatives of NumPy code.

Implementation of Graph Convolutional Networks in TensorFlow