All ML Frameworks

43,007 frameworks ranked by quality score · Page 2 of 431

Showing 101–200 of 43,007
# Framework Score Tier
101 openml/openml-python

OpenML's Python API for a World of Data and More ๐Ÿ’ซ

81
Verified
102 recommenders-team/recommenders

Best Practices on Recommendation Systems

81
Verified
103 PythonOT/POT

POT : Python Optimal Transport

81
Verified
104 lightly-ai/lightly

A python library for self-supervised learning on images.

81
Verified
105 davisking/dlib

A toolkit for making real world machine learning and data analysis...

81
Verified
106 zvtvz/zvt

modular quant framework.

81
Verified
107 talmolab/sleap

A deep learning framework for multi-animal pose tracking.

81
Verified
108 tensorflow/tfx

TFX is an end-to-end platform for deploying production ML pipelines

80
Verified
109 elyra-ai/elyra

Elyra extends JupyterLab with an AI centric approach.

80
Verified
110 timeseriesAI/tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai...

80
Verified
111 geomstats/geomstats

Computations and statistics on manifolds with geometric structures.

80
Verified
112 huggingface/evaluate

๐Ÿค— Evaluate: A library for easily evaluating machine learning models and datasets.

80
Verified
113 explosion/thinc

๐Ÿ”ฎ A refreshing functional take on deep learning, compatible with your...

80
Verified
114 mir-group/nequip

NequIP is a code for building E(3)-equivariant interatomic potentials

80
Verified
115 facebookresearch/fairseq2

FAIR Sequence Modeling Toolkit 2

80
Verified
116 feature-engine/feature_engine

Feature engineering and selection open-source Python library compatible with sklearn.

80
Verified
117 deepchem/deepchem

Democratizing Deep-Learning for Drug Discovery, Quantum Chemistry, Materials...

80
Verified
118 ndif-team/nnsight

The nnsight package enables interpreting and manipulating the internals of...

80
Verified
119 tensorflow/tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

80
Verified
120 tracel-ai/burn

Burn is a next generation tensor library and Deep Learning Framework that...

79
Verified
121 autogluon/autogluon

Fast and Accurate ML in 3 Lines of Code

79
Verified
122 MAIF/shapash

๐Ÿ”… Shapash: User-friendly Explainability and Interpretability to Develop...

79
Verified
123 alteryx/featuretools

An open source python library for automated feature engineering

79
Verified
124 JustGlowing/minisom

:red_circle: MiniSom is a minimalistic implementation of the Self Organizing Maps

79
Verified
125 winedarksea/AutoTS

Automated Time Series Forecasting

79
Verified
126 google/grain

Library for reading and processing ML training data.

79
Verified
127 replicate/cog

Containers for machine learning

79
Verified
128 webdataset/webdataset

A high-performance Python-based I/O system for large (and small) deep...

79
Verified
129 Trusted-AI/AIF360

A comprehensive set of fairness metrics for datasets and machine learning...

79
Verified
130 runpod/runpod-python

๐Ÿ | Python library for RunPod API and serverless worker SDK.

79
Verified
131 naptha/tesseract.js

Pure Javascript OCR for more than 100 Languages ๐Ÿ“–๐ŸŽ‰๐Ÿ–ฅ

78
Verified
132 supervisely/supervisely

Supervisely SDK for Python - convenient way to automate, customize and...

78
Verified
133 serengil/deepface

A Lightweight Face Recognition and Facial Attribute Analysis (Age, Gender,...

78
Verified
134 SimonBlanke/Gradient-Free-Optimizers

Lightweight optimization with local, global, population-based and sequential...

78
Verified
135 drivendataorg/cookiecutter-data-science

A logical, reasonably standardized, but flexible project structure for doing...

78
Verified
136 keras-team/keras-tuner

A Hyperparameter Tuning Library for Keras

78
Verified
137 py-why/dowhy

DoWhy is a Python library for causal inference that supports explicit...

78
Verified
138 trailofbits/fickling

A Python pickling decompiler and static analyzer

78
Verified
139 skfolio/skfolio

Python library for portfolio optimization built on top of scikit-learn

78
Verified
140 VowpalWabbit/vowpal_wabbit

Vowpal Wabbit is a machine learning system which pushes the frontier of...

78
Verified
141 Lightning-AI/LitServe

A minimal Python framework for building custom AI inference servers with...

78
Verified
142 CornellNLP/ConvoKit

ConvoKit is a toolkit for extracting conversational features and analyzing...

77
Verified
143 arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for...

77
Verified
144 fangwei123456/spikingjelly

SpikingJelly is an open-source deep learning framework for Spiking Neural...

77
Verified
145 Textualize/rich

Rich is a Python library for rich text and beautiful formatting in the terminal.

77
Verified
146 facebookresearch/habitat-lab

A modular high-level library to train embodied AI agents across a variety of...

77
Verified
147 horovod/horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

77
Verified
148 huggingface/autotrain-advanced

๐Ÿค— AutoTrain Advanced

77
Verified
149 sktime/skpro

A unified framework for tabular probabilistic regression, time-to-event...

77
Verified
150 pyRiemann/pyRiemann

Machine learning for multivariate data through the Riemannian geometry of...

76
Verified
151 apache/tvm

Open Machine Learning Compiler Framework

76
Verified
152 cleanlab/cleanlab

Cleanlab's open-source library is the standard data-centric AI package for...

76
Verified
153 birdnet-team/BirdNET-Analyzer

BirdNET analyzer for scientific audio data processing.

76
Verified
154 openvinotoolkit/openvino_notebooks

๐Ÿ“š Jupyter notebook tutorials for OpenVINOโ„ข

76
Verified
155 spring-projects/spring-ai

An Application Framework for AI Engineering

76
Verified
156 uxlfoundation/oneDNN

oneAPI Deep Neural Network Library (oneDNN)

76
Verified
157 DLR-RM/stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of...

76
Verified
158 NVIDIA/TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including...

76
Verified
159 infinitered/nsfwjs

NSFW detection on the client-side via TensorFlow.js

76
Verified
160 wkentaro/labelme

Image annotation with Python. Supports polygon, rectangle, circle, line,...

76
Verified
161 libAudioFlux/audioFlux

A library for audio and music analysis, feature extraction.

76
Verified
162 iree-org/iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

76
Verified
163 root-project/root

The official repository for ROOT: analyzing, storing and visualizing big...

76
Verified
164 opengeos/segment-geospatial

A Python package for segmenting geospatial data with the Segment Anything Model (SAM)

76
Verified
165 Xilinx/brevitas

Brevitas: neural network quantization in PyTorch

75
Verified
166 CederGroupHub/chgnet

Pretrained universal neural network potential for charge-informed atomistic...

75
Verified
167 flwrlabs/flower

Flower: A Friendly Federated AI Framework

75
Verified
168 datawhalechina/torch-rechub

A Lighting Pytorch Framework for Recommendation Models (PyTorchๆŽจ่็ฎ—ๆณ•ๆก†ๆžถ),...

75
Verified
169 stanfordmlgroup/ngboost

Natural Gradient Boosting for Probabilistic Prediction

75
Verified
170 hsahovic/poke-env

A python interface for training Reinforcement Learning bots to battle on...

75
Verified
171 seisbench/seisbench

SeisBench - A toolbox for machine learning in seismology

75
Verified
172 dusty-nv/jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

75
Verified
173 Cloud-CV/EvalAI

:cloud: :rocket: :bar_chart: :chart_with_upwards_trend: Evaluating state of...

75
Verified
174 haifengl/smile

Statistical Machine Intelligence & Learning Engine

75
Verified
175 rwth-i6/returnn

The RWTH extensible training framework for universal recurrent neural networks

75
Verified
176 EmuKit/emukit

A Python-based toolbox of various methods in decision making, uncertainty...

75
Verified
177 microsoft/mattersim

MatterSim: A deep learning atomistic model across elements, temperatures and...

74
Verified
178 a-r-j/graphein

Protein Graph Library

74
Verified
179 academic/awesome-datascience

:memo: An awesome Data Science repository to learn and apply for real world problems.

74
Verified
180 nmap/nmap

Nmap - the Network Mapper. Github mirror of official SVN repository.

74
Verified
181 Tencent/ncnn

ncnn is a high-performance neural network inference framework optimized for...

74
Verified
182 rasbt/mlxtend

A library of extension and helper modules for Python's data analysis and...

74
Verified
183 Fincept-Corporation/FinceptTerminal

FinceptTerminal is a modern finance application offering advanced market...

74
Verified
184 skops-dev/skops

skops is a Python library helping you share your scikit-learn based models...

74
Verified
185 awslabs/gluonts

Probabilistic time series modeling in Python

74
Verified
186 scikit-learn-contrib/imbalanced-learn

A Python Package to Tackle the Curse of Imbalanced Datasets in Machine Learning

74
Verified
187 neuromorphs/NIR

Neuromorphic Intermediate Representation reference implementation

74
Verified
188 google-research/google-research

Google Research

74
Verified
189 mmschlk/shapiq

Shapley Interactions and Shapley Values for Machine Learning

74
Verified
190 google-ai-edge/mediapipe

Cross-platform, customizable ML solutions for live and streaming media.

73
Verified
191 pycaret/pycaret

An open-source, low-code machine learning library in Python

73
Verified
192 raphaelvallat/antropy

AntroPy: entropy and complexity of (EEG) time-series in Python

73
Verified
193 sunlabuiuc/PyHealth

A Deep Learning Python Toolkit for Healthcare Applications.

73
Verified
194 keras-team/autokeras

AutoML library for deep learning

73
Verified
195 opengeos/geoai

GeoAI: Artificial Intelligence for Geospatial Data

73
Verified
196 yuce/pyswip

PySwip is a Python-Prolog interface that enables querying SWI-Prolog in your...

73
Verified
197 Denys88/rl_games

RL implementations

73
Verified
198 OpenMined/TenSEAL

A library for doing homomorphic encryption operations on tensors

73
Verified
199 TorchIO-project/torchio

Medical imaging processing for AI applications.

73
Verified
200 deepinv/deepinv

DeepInverse: a PyTorch library for solving imaging inverse problems using...

73
Verified