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

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

Deep Learning for humans

Tensors and Dynamic neural networks in Python with strong GPU acceleration

🐍 Geometric Computer Vision Library for Spatial AI

⚡ TabPFN: Foundation Model for Tabular Data ⚡

An Open Source Machine Learning Framework for Everyone

⛔️ DEPRECATED – See https://github.com/ageron/handson-ml3 or handson-mlp instead.

PArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice （『飞桨』核心框架，深度学习&机器学习高性能单机、分布式训练和跨平台部署）

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

💫 Industrial-strength Natural Language Processing (NLP) in Python

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

Datasets, Transforms and Models specific to Computer Vision

ChatterBot is a machine learning, conversational dialog engine for creating chat bots

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

NLTK Source