Machine Learning & Deep Learning Packages
Machine-learning and deep-learning frameworks and model libraries for training, fine-tuning, and serving models (PyTorch, scikit-learn, Transformers).
Packages in Machine Learning & Deep Learning
Kind2
Sort:Used by Most Apps
Transformers
State-of-the-art pretrained models for text, vision, audio, and multimodal inference and training.
huggingface_hub
The official Python client and hf CLI for the Hugging Face Hub, for downloading, uploading, and managing models, datasets, and Spaces.
qdrant-client
Official Python SDK for the Qdrant vector search engine
pymilvus
The official Python SDK for Milvus, the open-source vector database for embedding-based similarity search.
google-cloud-aiplatform
Google's official Python SDK for Vertex AI model training, deployment, and generative AI workflows
Gradio
Build and share machine learning web apps and demos in pure Python.
Replicate
Official JavaScript/TypeScript client for running and managing machine learning models on Replicate's cloud API.
wandb
Official Python SDK and CLI for the Weights & Biases ML experiment tracking platform
PySpark
Python API for Apache Spark, the unified engine for large-scale distributed data processing
PyTorch Lightning
A deep learning framework that organizes PyTorch training code and automates multi-GPU, mixed-precision, and distributed training.
MLflow
Open source AI engineering platform for tracking, evaluating, and shipping ML models, LLMs, and agents to production.
ModelScope
Alibaba's open model-hub library for NLP, CV, speech, and multi-modal AI models.
PyTorch Lightning
The deep learning framework that separates PyTorch research code from engineering boilerplate, scaling from CPU to multi-node GPU clusters without code changes.
dspy
DSPy replaces brittle prompt engineering with compositional Python code, letting optimizers tune your language model pipeline's prompts and weights automatically.
fastai
A high-level deep learning framework that layers state-of-the-art training APIs on top of PyTorch.
DeepEval
A Pytest-like open-source framework for unit-testing and evaluating LLM applications with research-backed metrics.
BasicSR
An open-source PyTorch toolbox for image and video restoration, covering super-resolution, denoising, deblurring, and face restoration.
ClearML
Python SDK for ML experiment tracking, dataset versioning, and MLOps pipeline orchestration.
GEPA
Optimize prompts, code, and agents with LLM reflection and evolutionary search
BrowserGym Core
Gymnasium environment for building and evaluating LLM-driven web agents in a real Chromium browser.
rasa-sdk
Python SDK for building the custom action server that connects a Rasa chatbot to your own APIs and business logic.
Label Studio SDK
The official Python client for the Label Studio API, covering projects, tasks, annotations, ML backends, and enterprise workspace management.
