All ML Frameworks

43,007 frameworks ranked by quality score

Showing 1–100 of 43,007

#	Framework	Score	Tier	Category	Stars	Language
1	microsoft/onnxruntime ONNX Runtime: cross-platform, high performance ML inferencing and training...	100	Verified	onnx-model-deployment	19,534	C++
2	PennyLaneAI/pennylane PennyLane is an open-source quantum software platform for quantum computing,...	100	Verified	quantum-machine-learning	3,105	Python
3	keras-team/keras Deep Learning for humans	100	Verified	lightweight-training-utilities	63,968	Python
4	huggingface/huggingface_hub The official Python client for the Hugging Face Hub.	100	Verified	huggingface-hub-clients	3,405	Python
5	PaddlePaddle/Paddle PArallel Distributed Deep LEarning: Machine Learning Framework from...	100	Verified	deep-learning-textbooks	23,752	C++
6	pytorch/pytorch Tensors and Dynamic neural networks in Python with strong GPU acceleration	100	Verified	lightweight-training-utilities	98,255	Python
7	opencv/opencv Open Source Computer Vision Library	100	Verified	computer-vision-frameworks	86,572	C++
8	scikit-learn/scikit-learn scikit-learn: machine learning in Python	100	Verified	—	65,422	Python
9	tensorflow/tensorflow An Open Source Machine Learning Framework for Everyone	100	Verified	platform-optimized-wheels	194,185	C++
10	onnx/onnx Open standard for machine learning interoperability	98	Verified	onnx-model-deployment	20,477	Python
11	dmlc/xgboost Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM)...	98	Verified	gradient-boosting-frameworks	28,121	C++
12	sktime/sktime A unified framework for machine learning with time series	98	Verified	time-series-forecasting	9,628	Python
13	catboost/catboost A fast, scalable, high performance Gradient Boosting on Decision Trees...	97	Verified	gradient-boosting-frameworks	8,841	C++
14	cvat-ai/cvat Annotate better with CVAT, the industry-leading data engine for machine...	96	Verified	data-annotation-tools	15,452	Python
15	online-ml/river 🌊 Online machine learning in Python	95	Verified	ml-drift-detection	5,746	Python
16	pytorch/executorch On-device AI across mobile, embedded and edge for PyTorch	95	Verified	tinyml-edge-deployment	4,374	Python
17	pytorch/vision Datasets, Transforms and Models specific to Computer Vision	95	Verified	computer-vision-frameworks	17,563	Python
18	streamlit/streamlit Streamlit — A faster way to build and share data apps.	95	Verified	streamlit-ml-dashboards	43,865	Python
19	Project-MONAI/MONAI AI Toolkit for Healthcare Imaging	94	Verified	medical-image-segmentation	7,927	Python
20	deepspeedai/DeepSpeed DeepSpeed is a deep learning optimization library that makes distributed...	94	Verified	distributed-training-frameworks	41,801	Python
21	Lightning-AI/pytorch-lightning Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero...	94	Verified	lightweight-training-utilities	30,923	Python
22	torchgeo/torchgeo TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data	94	Verified	geospatial-ml-tools	3,921	Python
23	kornia/kornia 🐍 Geometric Computer Vision Library for Spatial AI	93	Verified	computer-vision-frameworks	11,114	Python
24	pyg-team/pytorch_geometric Graph Neural Network Library for PyTorch	93	Verified	graph-neural-networks	23,561	Python
25	shap/shap A game theoretic approach to explain the output of any machine learning model.	92	Verified	shapley-value-methods	25,115	Jupyter Notebook
26	marimo-team/marimo A reactive notebook for Python — run reproducible experiments, query with...	92	Verified	ml-project-collections	19,647	Python
27	apple/coremltools Core ML tools contain supporting tools for Core ML model conversion,...	92	Verified	swift-ml-frameworks	5,182	Python
28	nilearn/nilearn Machine learning for NeuroImaging in Python	91	Verified	digital-pathology-mil	1,370	Python
29	aeon-toolkit/aeon A toolkit for time series machine learning and deep learning	91	Verified	time-series-forecasting	1,352	Python
30	dipy/dipy DIPY is the paragon 3D/4D+ medical imaging library in Python. Contains...	91	Verified	medical-image-segmentation	811	Python
31	HumanSignal/label-studio Label Studio is a multi-type data labeling and annotation tool with...	91	Verified	data-annotation-tools	26,703	TypeScript
32	Nixtla/neuralforecast Scalable and user friendly neural :brain: forecasting algorithms.	91	Verified	time-series-forecasting	4,003	Python
33	tslearn-team/tslearn The machine learning toolkit for time series analysis in Python	91	Verified	time-series-forecasting	3,125	Python
34	biolab/orange3 🍊 :bar_chart: :bulb: Orange: Interactive data analysis	90	Verified	data-quality-preprocessing	5,573	Python
35	modelscope/modelscope ModelScope: bring the notion of Model-as-a-Service to life.	90	Verified	model-inference-serving	8,784	Python
36	pallets/click Python composable command line interface toolkit	90	Verified	—	17,355	Python
37	carla-simulator/carla Open-source simulator for autonomous driving research.	90	Verified	autonomous-driving-projects	13,659	C++
38	lmcinnes/umap Uniform Manifold Approximation and Projection	90	Verified	manifold-learning-frameworks	8,114	Python
39	voicepaw/so-vits-svc-fork so-vits-svc fork with realtime support, improved interface and more features.	89	Verified	text-to-speech-frameworks	9,281	Python
40	uxlfoundation/scikit-learn-intelex Extension for Scikit-learn is a seamless way to speed up your Scikit-learn...	89	Verified	ml-algorithm-implementations	1,334	Python
41	Nixtla/statsforecast Lightning ⚡️ fast forecasting with statistical and econometric models.	89	Verified	time-series-forecasting	4,718	Python
42	shankarpandala/lazypredict Lazy Predict help build a lot of basic models without much code and helps...	89	Verified	automl-frameworks	3,301	Python
43	casadi/casadi CasADi is a symbolic framework for numeric optimization implementing...	89	Verified	robotics-control-optimization	2,163	C++
44	deepmodeling/deepmd-kit A deep learning package for many-body potential energy representation and...	89	Verified	chemical-property-ml	1,892	Python
45	voxel51/fiftyone Refine high-quality datasets and visual AI models	89	Verified	ml-frameworks-directories	10,447	Python
46	scverse/scanpy Single-cell analysis in Python. Scales to >100M cells.	88	Verified	single-cell-omics-integration	2,367	Python
47	scverse/anndata Annotated data.	88	Verified	scikit-learn-pipelines	720	Python
48	scverse/scvi-tools Deep probabilistic analysis of single-cell and spatial omics data	88	Verified	single-cell-omics-integration	1,582	Python
49	brian-team/brian2 Brian is a free, open source simulator for spiking neural networks.	88	Verified	robotics-control-optimization	1,142	Python
50	Lightning-AI/torchmetrics Machine learning metrics for distributed, scalable PyTorch applications.	88	Verified	lightweight-training-utilities	2,418	Python
51	meta-pytorch/torchrec Pytorch domain library for recommendation systems	88	Verified	recommendation-system-frameworks	2,488	Python
52	process-intelligence-solutions/pm4py Official public repository for PM4Py (Process Mining for Python) — an...	88	Verified	automl-frameworks	926	Python
53	fastai/fastai The fastai deep learning library	88	Verified	deep-learning-textbooks	27,903	Jupyter Notebook
54	mne-tools/mne-python MNE: Magnetoencephalography (MEG) and Electroencephalography (EEG) in Python	88	Verified	eeg-brain-signal-processing	3,284	Python
55	uber/causalml Uplift modeling and causal inference with machine learning algorithms	88	Verified	causal-inference-ml	5,758	Python
56	pymc-devs/pytensor PyTensor allows you to define, optimize, and efficiently evaluate...	87	Verified	tensor-algebra-frameworks	598	Python
57	sebp/scikit-survival Survival analysis built on top of scikit-learn	87	Verified	survival-analysis-ml	1,282	Python
58	microsoft/FLAML A fast library for AutoML and tuning. Join our Discord:...	87	Verified	automl-frameworks	4,298	Jupyter Notebook
59	devitocodes/devito DSL and compiler framework for automated finite-differences and stencil computation	87	Verified	robotics-control-optimization	679	Python
60	terrastackai/terratorch A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).	87	Verified	geospatial-ml-tools	749	Python
61	unit8co/darts A python library for user-friendly forecasting and anomaly detection on time series.	87	Verified	anomaly-detection-systems	9,248	Python
62	NVIDIA/TensorRT NVIDIA® TensorRT™ is an SDK for high-performance deep learning inference on...	87	Verified	onnx-model-deployment	12,784	C++
63	probabl-ai/skore Track your Data Science. Skore's open-source Python library accelerates ML...	87	Verified	—	601	Python
64	Farama-Foundation/Gymnasium An API standard for single-agent reinforcement learning environments, with...	87	Verified	—	11,640	Python
65	scikit-learn-contrib/hdbscan A high performance implementation of HDBSCAN clustering.	87	Verified	clustering-algorithm-implementations	3,080	Jupyter Notebook
66	lanpa/tensorboardX tensorboard for pytorch (and chainer, mxnet, numpy, ...)	87	Verified	lightweight-training-utilities	7,988	Python
67	opentensor/bittensor Internet-scale Neural Networks	86	Verified	ml-benchmarking-frameworks	1,383	Python
68	NeuroTechX/moabb Mother of All BCI Benchmarks	86	Verified	eeg-brain-signal-processing	944	Python
69	open-edge-platform/datumaro Dataset Management Framework, a Python library and a CLI tool to build,...	86	Verified	open-dataset-collections	661	Python
70	netket/netket Machine learning algorithms for many-body quantum systems	86	Verified	quantum-machine-learning	668	Python
71	skforecast/skforecast Time series forecasting with machine learning models	86	Verified	time-series-forecasting	1,462	Python
72	mlfoundations/open_clip An open source implementation of CLIP.	86	Verified	clip-vision-language	13,496	Python
73	google-deepmind/dm_control Google DeepMind's software stack for physics-based simulation and...	86	Verified	reinforcement-learning-frameworks	4,494	Python
74	google-deepmind/optax Optax is a gradient processing and optimization library for JAX.	85	Verified	jax-ml-frameworks	2,207	Python
75	treeverse/dvc 🦉 Data Versioning and ML Experiments	85	Verified	ml-experiment-tracking	15,443	Python
76	NVIDIA/earth2studio Open-source deep-learning framework for exploring, building and deploying AI...	85	Verified	weather-forecasting-ml	694	Python
77	huggingface/datasets 🤗 The largest hub of ready-to-use datasets for AI models with fast,...	85	Verified	llm-data-labeling	21,273	Python
78	roboflow/inference Turn any computer or edge device into a command center for your computer...	85	Verified	video-processing-pipelines	2,217	Python
79	optuna/optuna A hyperparameter optimization framework	85	Verified	hyperparameter-optimization	13,673	Python
80	braindecode/braindecode Deep learning software to decode EEG, ECG or MEG signals	85	Verified	eeg-brain-signal-processing	1,175	Python
81	PINTO0309/onnx2tf Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras...	84	Verified	onnx-model-deployment	931	Python
82	skrub-data/skrub Machine learning with dataframes	84	Verified	data-quality-preprocessing	1,578	Python
83	patrick-kidger/equinox Elegant easy-to-use neural networks + scientific computing in JAX....	84	Verified	jax-ml-frameworks	2,814	Python
84	TissueImageAnalytics/tiatoolbox Computational Pathology Toolbox developed by TIA Centre, University of Warwick.	84	Verified	digital-pathology-mil	513	Python
85	fairlearn/fairlearn A Python package to assess and improve fairness of machine learning models.	84	Verified	fairness-bias-auditing	2,213	Python
86	InterDigitalInc/CompressAI A PyTorch library and evaluation platform for end-to-end compression research	84	Verified	neural-data-compression	1,535	Python
87	py-why/EconML ALICE (Automated Learning and Intelligence for Causation and Economics) is a...	84	Verified	causal-inference-ml	4,537	Jupyter Notebook
88	gradio-app/gradio Build and share delightful machine learning apps, all in Python. 🌟 Star to...	84	Verified	segment-anything-applications	42,013	Python
89	qiskit-community/qiskit-machine-learning An open-source library built on Qiskit for quantum machine learning tasks at...	83	Verified	quantum-machine-learning	951	Python
90	sbi-dev/sbi sbi is a Python package for simulation-based inference, designed to meet the...	83	Verified	gaussian-process-frameworks	801	Python
91	PriorLabs/TabPFN ⚡ TabPFN: Foundation Model for Tabular Data ⚡	83	Verified	tabular-deep-learning	5,846	Python
92	joblib/joblib Computing with Python functions.	83	Verified	—	4,332	Python
93	Nixtla/mlforecast Scalable machine 🤖 learning for time series forecasting.	82	Verified	time-series-forecasting	1,186	Python
94	CVHub520/X-AnyLabeling Effortless data labeling with AI support from Segment Anything and other...	82	Verified	data-annotation-tools	8,375	Python
95	google/magika Fast and accurate AI powered file content types detection	82	Verified	ai-generated-image-detection	10,151	Python
96	microsoft/tensorwatch Debugging, monitoring and visualization for Python Machine Learning and Data Science	82	Verified	platform-optimized-wheels	3,466	Jupyter Notebook
97	encode/httpx A next generation HTTP client for Python. 🦋	82	Verified	—	15,147	Python
98	omni-us/jsonargparse Minimal effort CLIs derived from type hints and parse from command line,...	82	Verified	—	423	Python
99	obss/sahi Framework agnostic sliced/tiled inference + interactive ui + error analysis plots	81	Verified	explainability-interpretability-frameworks	5,160	Python
100	chemprop/chemprop Message Passing Neural Networks for Molecule Property Prediction	81	Verified	chemical-property-ml	2,289	Python

1 2 3 … … 429 430 431 Next »