PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Machine Learning Python Packages

Python packages with the GitHub topic machine-learning. Sorted by relevance, with stars and monthly downloads.
huggingface
huggingface-hub

The official Python client for the Hugging Face Hub.

256.1M 4K 1K
scikit-learn
scikit-learn

scikit-learn: machine learning in Python

208.6M 66K 27K
huggingface
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.

149.1M 161K 33K
huggingface
datasets

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

127.8M 22K 3K
pytorch
torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

87.9M 100K 28K
microsoft
onnxruntime

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

71.1M 21K 4K
nltk
nltk

NLTK Source

63.2M 15K 3K
ray-project
ray

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

56.1M 43K 8K
modal-labs
modal

SDK libraries for Modal

51.3M 473 95
apache
apache-airflow-providers-common-sql

Apache Airflow - A platform to programmatically author, schedule, and monitor workflows

45.9M 45K 17K
dmlc
xgboost

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

43.1M 28K 9K
pytorch
torchvision

Datasets, Transforms and Models specific to Computer Vision

38.4M 18K 7K
mlflow
mlflow-skinny

The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, and optimize production-quality AI applications while controlling costs and managing access to models and data.

38.2M 26K 6K
mlflow
mlflow

The open source AI engineering platform for agents, LLMs, and ML models. MLflow enables teams of all sizes to debug, evaluate, monitor, and optimize production-quality AI applications while controlling costs and managing access to models and data.

37.1M 26K 6K
apache
apache-airflow-providers-fab

Apache Airflow - A platform to programmatically author, schedule, and monitor workflows

29.9M 45K 17K
supabase
realtime

Python Client for Supabase. Query Postgres from Flask, Django, FastAPI. Python user authentication, security policies, edge functions, file storage, and realtime data streaming. Good first issue.

29.5M 3K 484
streamlit
streamlit

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

29.2M 45K 4K
wandb
wandb

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

26.2M 11K 872
supabase
supabase

Python Client for Supabase. Query Postgres from Flask, Django, FastAPI. Python user authentication, security policies, edge functions, file storage, and realtime data streaming. Good first issue.

25.1M 3K 484
explosion
thinc

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

24.4M 3K 292
supabase
storage3

Python Client for Supabase. Query Postgres from Flask, Django, FastAPI. Python user authentication, security policies, edge functions, file storage, and realtime data streaming. Good first issue.

23.7M 3K 484
supabase
postgrest

Python Client for Supabase. Query Postgres from Flask, Django, FastAPI. Python user authentication, security policies, edge functions, file storage, and realtime data streaming. Good first issue.

23.3M 3K 484
apache
apache-airflow-providers-http

Apache Airflow - A platform to programmatically author, schedule, and monitor workflows

22.6M 45K 17K
tensorflow
tensorflow

An Open Source Machine Learning Framework for Everyone

22.6M 195K 75K
    • Data from PyPI, GitHub, ClickHouse, and BigQuery