jax by jax-ml

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

updated at Nov. 17, 2024, 2:12 a.m.

337 +3

30,510 +74

2,802 +7

pytorch-lightning by PyTorchLightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.

updated at Nov. 17, 2024, 2:09 a.m.

250 +1

28,392 +45

3,384 +1

LightGBM by Microsoft

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

updated at Nov. 17, 2024, 2:09 a.m.

434 +0

16,698 +19

3,834 +1

albumentations by albumentations-team

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

updated at Nov. 17, 2024, 1:30 a.m.

129 +1

14,258 +30

1,649 +2

vizro by mckinsey

Vizro is a toolkit for creating modular data visualization applications.

updated at Nov. 17, 2024, 1:29 a.m.

18 +0

2,705 +19

142 +0

statsforecast by Nixtla

Lightning ⚡️ fast forecasting with statistical and econometric models.

updated at Nov. 17, 2024, 1:17 a.m.

41 -1

3,988 +17

283 +1

graph_nets by deepmind

Build Graph Nets in Tensorflow

updated at Nov. 17, 2024, 1:17 a.m.

225 +0

5,360 +1

782 +0

pycaret by pycaret

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

updated at Nov. 17, 2024, 1:09 a.m.

136 +1

8,955 +26

1,774 +3

geopandas by geopandas

Python tools for geographic data

updated at Nov. 17, 2024, 1:08 a.m.

104 +0

4,519 +9

935 +1

auto-sklearn by automl

Automated Machine Learning with scikit-learn

updated at Nov. 17, 2024, 12:58 a.m.

215 +0

7,634 +13

1,280 +1

ludwig by ludwig-ai

Low-code framework for building custom LLMs, neural networks, and other AI models

updated at Nov. 17, 2024, 12:45 a.m.

194 +0

11,186 +7

1,194 +2

pytorch_geometric by pyg-team

Graph Neural Network Library for PyTorch

updated at Nov. 17, 2024, 12:20 a.m.

252 +0

21,387 +49

3,670 +12

tensorflow by tensorflow

An Open Source Machine Learning Framework for Everyone

updated at Nov. 17, 2024, 12:04 a.m.

7,573 -5

186,403 +124

74,304 -1

skpro by sktime

A unified framework for tabular probabilistic regression, time-to-event prediction, and probability distributions in python

updated at Nov. 16, 2024, 11:56 p.m.

11 -1

249 +1

46 +1

PettingZoo by Farama-Foundation

An API standard for multi-agent reinforcement learning environments, with popular reference environments and related utilities

updated at Nov. 16, 2024, 11:50 p.m.

18 +0

2,624 +7

414 +1

stellargraph by stellargraph

StellarGraph - Machine Learning on Graphs

updated at Nov. 16, 2024, 11:47 p.m.

62 +0

2,948 +1

431 +0

pytorch by pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

updated at Nov. 16, 2024, 11:30 p.m.

1,743 +1

83,990 +192

22,642 +50

sktime by sktime

A unified framework for machine learning with time series

updated at Nov. 16, 2024, 10:53 p.m.

108 +0

7,947 +19

1,375 +6

matplotlib by matplotlib

matplotlib: plotting with Python

updated at Nov. 16, 2024, 10:49 p.m.

592 +0

20,302 +30

7,653 +4

mlpack by mlpack

mlpack: a fast, header-only C++ machine learning library

updated at Nov. 16, 2024, 10:05 p.m.

184 +0

5,117 +11

1,607 +2