Artificial Intelligence Python Packages

datasets

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

thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

spacy

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

onnx

Open standard for machine learning interoperability

pytorch-lightning

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

elevenlabs

The official Python SDK for the ElevenLabs API.

lightning

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

kornia

🐍 Geometric Computer Vision Library for Spatial AI

semantic-kernel

Integrate cutting-edge LLM technology quickly and easily into your apps

marimo

A reactive notebook for Python — run reproducible experiments, query with SQL, execute as a script, deploy as an app, and version with git. Stored as pure Python. All in a modern, AI-native editor.

prometheus-api-client

A python wrapper for the prometheus http api

gluonts

Probabilistic time series modeling in Python

sklearn-compat

Ease multi-version support for scikit-learn compatible library

vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch

x-transformers

A concise but complete full-attention transformer with a set of promising experimental features from various papers

openhands-ai

🙌 OpenHands: AI-Driven Development

runpod

🐍 | Python library for RunPod API and serverless worker SDK.

interpret-core

Fit interpretable models. Explain blackbox machine learning.

stanza

Stanford NLP Python library for tokenization, sentence segmentation, NER, and parsing of many human languages

daft

High-performance data engine for AI and multimodal workloads. Process images, audio, video, and structured data at any scale

truss

The simplest way to serve AI/ML models in production

rotary-embedding-torch

Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch

chronos-forecasting

Chronos: Pretrained Models for Time Series Forecasting

torch-einops-utils

Some utility functions to help myself (and perhaps others) go faster with ML/AI work