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.

lightning

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

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.

kornia

🐍 Geometric Computer Vision Library for Spatial AI

sklearn-compat

Ease multi-version support for scikit-learn compatible library

prometheus-api-client

A python wrapper for the prometheus http api

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

chronos-forecasting

Chronos: Pretrained Models for Time Series Forecasting

interpret-core

Fit interpretable models. Explain blackbox machine learning.

runpod

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

openhands-ai

🙌 OpenHands: AI-Driven Development

torch-einops-utils

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

vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch

baseten-performance-client

The simplest way to serve AI/ML models in production

nautilus-trader

Production-grade Rust-native trading engine with deterministic event-driven architecture

conformer

Implementation of the convolutional module from the Conformer paper, for use in Transformers

truss

The simplest way to serve AI/ML models in production

semantic-kernel

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