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

spacy

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

thinc

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

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.

openhands-ai

🙌 OpenHands: AI-Driven Development

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

prometheus-api-client

A python wrapper for the prometheus http api

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.

sklearn-compat

Ease multi-version support for scikit-learn compatible library

x-transformers

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

vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch

runpod

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

truss

The simplest way to serve AI/ML models in production

daft

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

stanza

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

interpret-core

Fit interpretable models. Explain blackbox machine learning.

gluonts

Probabilistic time series modeling in Python

rotary-embedding-torch

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

torch-einops-utils

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

interpret

Fit interpretable models. Explain blackbox machine learning.