Deep Learning Python Packages

huggingface-hub

The official CLI and Python client for the Hugging Face Hub.

transformers

🤗 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.

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

datasets

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

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

spacy

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

thinc

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

streamlit

Streamlit — A faster way to build and share data apps.

magika

Fast and accurate AI powered file content types detection

wandb

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

ctranslate2

Fast inference engine for Transformer models

ray

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

onnx

Open standard for machine learning interoperability

tensorflow

An Open Source Machine Learning Framework for Everyone

keras

Deep Learning for humans

torchmetrics

Machine learning metrics for distributed, scalable PyTorch applications.

faster-whisper

Faster Whisper transcription with CTranslate2

pytorch-lightning

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

shap

A game theoretic approach to explain the output of any machine learning model.

gradio-client

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

gradio

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

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

nvidia-cutlass-dsl-libs-base

CUDA Templates and Python DSLs for High-Performance Linear Algebra