All ML Frameworks

45,958 frameworks ranked by quality score

Showing 1–100 of 45,958

#	Framework	Score	Tier	Category	Stars	Language
1	fastapi/fastapi FastAPI framework, high performance, easy to learn, fast to code, ready for...	95	Verified	—	97,109	Python
2	microsoft/onnxruntime ONNX Runtime: cross-platform, high performance ML inferencing and training...	93	Verified	onnx-model-deployment	19,534	C++
3	PennyLaneAI/pennylane PennyLane is an open-source quantum software platform for quantum computing,...	87	Verified	quantum-machine-learning	3,105	Python
4	casadi/casadi CasADi is a symbolic framework for numeric optimization implementing...	87	Verified	robotics-control-optimization	2,163	C++
5	pytorch/pytorch Tensors and Dynamic neural networks in Python with strong GPU acceleration	87	Verified	lightweight-training-utilities	98,255	Python
6	mne-tools/mne-python MNE: Magnetoencephalography (MEG) and Electroencephalography (EEG) in Python	87	Verified	eeg-brain-signal-processing	3,284	Python
7	huggingface/huggingface_hub The official Python client for the Hugging Face Hub.	87	Verified	huggingface-hub-clients	3,405	Python
8	PaddlePaddle/Paddle PArallel Distributed Deep LEarning: Machine Learning Framework from...	87	Verified	deep-learning-textbooks	23,752	C++
9	tensorflow/tensorflow An Open Source Machine Learning Framework for Everyone	87	Verified	platform-optimized-wheels	194,185	C++
10	scikit-learn/scikit-learn scikit-learn: machine learning in Python	87	Verified	—	65,422	Python
11	keras-team/keras Deep Learning for humans	87	Verified	lightweight-training-utilities	63,968	Python
12	dipy/dipy DIPY is the paragon 3D/4D+ medical imaging library in Python. Contains...	85	Verified	medical-image-segmentation	811	Python
13	devitocodes/devito DSL and compiler framework for automated finite-differences and stencil computation	85	Verified	robotics-control-optimization	679	Python
14	sktime/sktime A unified framework for machine learning with time series	85	Verified	time-series-forecasting	9,628	Python
15	brian-team/brian2 Brian is a free, open source simulator for spiking neural networks.	85	Verified	robotics-control-optimization	1,142	Python
16	nilearn/nilearn Machine learning for NeuroImaging in Python	85	Verified	digital-pathology-mil	1,370	Python
17	aeon-toolkit/aeon A toolkit for time series machine learning and deep learning	85	Verified	time-series-forecasting	1,352	Python
18	opencv/opencv Open Source Computer Vision Library	85	Verified	computer-vision-frameworks	86,572	C++
19	dmlc/xgboost Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM)...	85	Verified	gradient-boosting-frameworks	28,121	C++
20	onnx/onnx Open standard for machine learning interoperability	85	Verified	onnx-model-deployment	20,477	Python
21	deepmodeling/deepmd-kit A deep learning package for many-body potential energy representation and...	84	Verified	chemical-property-ml	1,892	Python
22	Project-MONAI/MONAI AI Toolkit for Healthcare Imaging	84	Verified	medical-image-segmentation	7,927	Python
23	catboost/catboost A fast, scalable, high performance Gradient Boosting on Decision Trees...	84	Verified	gradient-boosting-frameworks	8,841	C++
24	pymc-devs/pytensor PyTensor allows you to define, optimize, and efficiently evaluate...	83	Verified	tensor-algebra-frameworks	598	Python
25	probabl-ai/skore Track your Data Science. Skore's open-source Python library accelerates ML...	83	Verified	—	601	Python
26	cvat-ai/cvat Annotate better with CVAT, the industry-leading data engine for machine...	83	Verified	data-annotation-tools	15,452	Python
27	uxlfoundation/scikit-learn-intelex Extension for Scikit-learn is a seamless way to speed up your Scikit-learn...	83	Verified	ml-algorithm-implementations	1,334	Python
28	braindecode/braindecode Deep learning software to decode EEG, ECG or MEG signals	82	Verified	eeg-brain-signal-processing	1,175	Python
29	shap/shap A game theoretic approach to explain the output of any machine learning model.	82	Verified	shapley-value-methods	25,115	Jupyter Notebook
30	huggingface/datasets 🤗 The largest hub of ready-to-use datasets for AI models with fast,...	82	Verified	llm-data-labeling	21,273	Python
31	online-ml/river 🌊 Online machine learning in Python	82	Verified	ml-drift-detection	5,746	Python
32	Lightning-AI/torchmetrics Machine learning metrics for distributed, scalable PyTorch applications.	82	Verified	lightweight-training-utilities	2,418	Python
33	pytorch/executorch On-device AI across mobile, embedded and edge for PyTorch	82	Verified	tinyml-edge-deployment	4,374	Python
34	optuna/optuna A hyperparameter optimization framework	82	Verified	hyperparameter-optimization	13,673	Python
35	meta-pytorch/torchrec Pytorch domain library for recommendation systems	82	Verified	recommendation-system-frameworks	2,488	Python
36	scverse/anndata Annotated data.	82	Verified	scikit-learn-pipelines	720	Python
37	process-intelligence-solutions/pm4py Official public repository for PM4Py (Process Mining for Python) — an...	82	Verified	automl-frameworks	926	Python
38	pytorch/vision Datasets, Transforms and Models specific to Computer Vision	82	Verified	computer-vision-frameworks	17,563	Python
39	scverse/scvi-tools Deep probabilistic analysis of single-cell and spatial omics data	82	Verified	single-cell-omics-integration	1,582	Python
40	scverse/scanpy Single-cell analysis in Python. Scales to >100M cells.	82	Verified	single-cell-omics-integration	2,367	Python
41	streamlit/streamlit Streamlit — A faster way to build and share data apps.	82	Verified	streamlit-ml-dashboards	43,865	Python
42	gradio-app/gradio Build and share delightful machine learning apps, all in Python. 🌟 Star to...	81	Verified	segment-anything-applications	42,013	Python
43	deepspeedai/DeepSpeed DeepSpeed is a deep learning optimization library that makes distributed...	81	Verified	distributed-training-frameworks	41,801	Python
44	torchgeo/torchgeo TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data	81	Verified	geospatial-ml-tools	3,921	Python
45	Nixtla/neuralforecast Scalable and user friendly neural :brain: forecasting algorithms.	81	Verified	time-series-forecasting	4,003	Python
46	Farama-Foundation/Gymnasium An API standard for single-agent reinforcement learning environments, with...	81	Verified	—	11,640	Python
47	Lightning-AI/pytorch-lightning Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero...	81	Verified	lightweight-training-utilities	30,923	Python
48	sebp/scikit-survival Survival analysis built on top of scikit-learn	81	Verified	survival-analysis-ml	1,282	Python
49	kornia/kornia 🐍 Geometric Computer Vision Library for Spatial AI	80	Verified	computer-vision-frameworks	11,114	Python
50	WMD-group/SMACT Python package to aid materials design and informatics	80	Verified	—	133	Python
51	sbi-dev/sbi sbi is a Python package for simulation-based inference, designed to meet the...	80	Verified	gaussian-process-frameworks	801	Python
52	open-edge-platform/datumaro Dataset Management Framework, a Python library and a CLI tool to build,...	80	Verified	open-dataset-collections	661	Python
53	qiskit-community/qiskit-machine-learning An open-source library built on Qiskit for quantum machine learning tasks at...	80	Verified	quantum-machine-learning	951	Python
54	PriorLabs/TabPFN ⚡ TabPFN: Foundation Model for Tabular Data ⚡	80	Verified	tabular-deep-learning	5,846	Python
55	NeuroTechX/moabb Mother of All BCI Benchmarks	80	Verified	eeg-brain-signal-processing	944	Python
56	opentensor/bittensor Internet-scale Neural Networks	80	Verified	ml-benchmarking-frameworks	1,383	Python
57	NVIDIA/earth2studio Open-source deep-learning framework for exploring, building and deploying AI...	80	Verified	weather-forecasting-ml	694	Python
58	pyg-team/pytorch_geometric Graph Neural Network Library for PyTorch	80	Verified	graph-neural-networks	23,561	Python
59	recommenders-team/recommenders Best Practices on Recommendation Systems	80	Verified	recommendation-system-frameworks	21,514	Python
60	explosion/thinc 🔮 A refreshing functional take on deep learning, compatible with your...	80	Verified	jax-ml-frameworks	2,893	Python
61	lmcinnes/umap Uniform Manifold Approximation and Projection	80	Verified	manifold-learning-frameworks	8,114	Python
62	pallets/click Python composable command line interface toolkit	80	Verified	—	17,355	Python
63	pyRiemann/pyRiemann Machine learning for multivariate data through the Riemannian geometry of...	80	Verified	manifold-learning-frameworks	743	Python
64	google-deepmind/optax Optax is a gradient processing and optimization library for JAX.	79	Verified	jax-ml-frameworks	2,207	Python
65	terrastackai/terratorch A Python toolkit for fine-tuning Geospatial Foundation Models (GFMs).	79	Verified	geospatial-ml-tools	749	Python
66	yzhao062/pyod A Python Library for Outlier and Anomaly Detection, Integrating Classical...	79	Verified	anomaly-detection-systems	9,747	Python
67	roboflow/inference Turn any computer or edge device into a command center for your computer...	79	Verified	video-processing-pipelines	2,217	Python
68	marimo-team/marimo A reactive notebook for Python — run reproducible experiments, query with...	79	Verified	ml-project-collections	19,647	Python
69	aramis-lab/clinica Software platform for clinical neuroimaging studies	78	Verified	digital-pathology-mil	272	Python
70	PINTO0309/onnx2tf Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras...	78	Verified	onnx-model-deployment	931	Python
71	py-why/dowhy DoWhy is a Python library for causal inference that supports explicit...	78	Verified	causal-inference-ml	7,995	Python
72	tslearn-team/tslearn The machine learning toolkit for time series analysis in Python	78	Verified	time-series-forecasting	3,125	Python
73	fastai/fastai The fastai deep learning library	78	Verified	deep-learning-textbooks	27,903	Jupyter Notebook
74	davisking/dlib A toolkit for making real world machine learning and data analysis...	78	Verified	cpp-ml-libraries	14,366	C++
75	py-why/EconML ALICE (Automated Learning and Intelligence for Causation and Economics) is a...	78	Verified	causal-inference-ml	4,537	Jupyter Notebook
76	geomstats/geomstats Computations and statistics on manifolds with geometric structures.	78	Verified	manifold-learning-frameworks	1,465	Python
77	fairlearn/fairlearn A Python package to assess and improve fairness of machine learning models.	78	Verified	fairness-bias-auditing	2,213	Python
78	skrub-data/skrub Machine learning with dataframes	78	Verified	data-quality-preprocessing	1,578	Python
79	voicepaw/so-vits-svc-fork so-vits-svc fork with realtime support, improved interface and more features.	78	Verified	text-to-speech-frameworks	9,281	Python
80	chemprop/chemprop Message Passing Neural Networks for Molecule Property Prediction	78	Verified	chemical-property-ml	2,289	Python
81	netket/netket Machine learning algorithms for many-body quantum systems	78	Verified	quantum-machine-learning	668	Python
82	HumanSignal/label-studio Label Studio is a multi-type data labeling and annotation tool with...	78	Verified	data-annotation-tools	26,703	TypeScript
83	tensorflow/tfx TFX is an end-to-end platform for deploying production ML pipelines	78	Verified	scala-ml-frameworks	2,174	Python
84	brainglobe/cellfinder Automated 3D cell detection in very large images	78	Verified	—	226	Python
85	scikit-learn-contrib/hdbscan A high performance implementation of HDBSCAN clustering.	77	Verified	clustering-algorithm-implementations	3,080	Jupyter Notebook
86	modelscope/modelscope ModelScope: bring the notion of Model-as-a-Service to life.	77	Verified	model-inference-serving	8,784	Python
87	Acellera/moleculekit MoleculeKit: Your favorite molecule manipulation kit	77	Verified	chemical-property-ml	236	Python
88	TorchIO-project/torchio Medical imaging processing for AI applications.	77	Verified	data-augmentation-techniques	2,370	Python
89	mir-group/nequip NequIP is a code for building E(3)-equivariant interatomic potentials	77	Verified	chemical-property-ml	878	Python
90	deepinv/deepinv DeepInverse: a PyTorch library for solving imaging inverse problems using...	77	Verified	semantic-segmentation-techniques	674	Python
91	lanpa/tensorboardX tensorboard for pytorch (and chainer, mxnet, numpy, ...)	77	Verified	lightweight-training-utilities	7,988	Python
92	Textualize/rich Rich is a Python library for rich text and beautiful formatting in the terminal.	77	Verified	—	55,962	Python
93	datawhalechina/torch-rechub A Lighting Pytorch Framework for Recommendation Models (PyTorch推荐算法框架),...	77	Verified	llm-recommendation-systems	930	Jupyter Notebook
94	carla-simulator/carla Open-source simulator for autonomous driving research.	77	Verified	autonomous-driving-projects	13,659	C++
95	scikit-learn-contrib/imbalanced-learn A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning	77	Verified	class-imbalance-solutions	7,090	Python
96	sunlabuiuc/PyHealth A Deep Learning Python Toolkit for Healthcare Applications.	77	Verified	healthcare-ai-applications	1,486	Python
97	biolab/orange3 🍊 :bar_chart: :bulb: Orange: Interactive data analysis	77	Verified	data-quality-preprocessing	5,573	Python
98	tracel-ai/burn Burn is a next generation tensor library and Deep Learning Framework that...	76	Verified	rust-onnx-runtime	14,588	Rust
99	bbfamily/abu 阿布量化交易系统(股票，期权，期货，比特币，机器学习) 基于python的开源量化交易，量化投资架构	76	Verified	quantitative-trading-frameworks	16,451	Python
100	replicate/cog Containers for machine learning	76	Verified	ml-development-environments	9,268	Go

1 2 3 … … 458 459 460 Next »