GreatX by EdisonLeeeee

A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).

created at Nov. 15, 2021, 1:06 a.m.

7 +0

80 +0

11 +0

rlpyt by astooke

Reinforcement Learning in PyTorch

created at April 28, 2019, 6:29 p.m.

52 +0

2,206 +1

324 +0

pytorch_geometric_signed_directed by SherylHYX

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

created at Oct. 28, 2021, 4:26 a.m.

7 +0

114 +0

15 +0

machin by iffiX

Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...

created at June 7, 2020, 5:13 a.m.

4 +0

390 +0

51 +0

autokeras by keras-team

AutoML library for deep learning

created at Nov. 19, 2017, 11:18 p.m.

302 +0

9,090 +3

1,400 +1

mmengine by open-mmlab

OpenMMLab Foundational Library for Training Deep Learning Models

created at Feb. 8, 2022, 9:05 a.m.

28 +0

1,072 +7

323 +1

treelite by dmlc

Universal model exchange and serialization format for decision tree forests

created at June 28, 2017, 11:39 p.m.

40 +0

717 +1

96 +0

allRank by allegro

allRank is a framework for training learning-to-rank neural models based on PyTorch.

created at Oct. 10, 2019, 1:39 p.m.

28 +1

816 +2

114 +0

RecBole by RUCAIBox

A unified, comprehensive and efficient recommendation library

created at June 11, 2020, 3:18 p.m.

40 +0

3,230 +15

588 +1

PettingZoo by Farama-Foundation

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

created at Jan. 20, 2020, 3:41 a.m.

20 +0

2,427 +16

395 +2

ranking by tensorflow

Learning to Rank in TensorFlow

created at Dec. 3, 2018, 8:48 p.m.

97 +0

2,719 +0

473 +0

vizro by mckinsey

Vizro is a toolkit for creating modular data visualization applications.

created at Sept. 4, 2023, 11:11 a.m.

18 +0

2,442 +6

107 +0

recommenders by tensorflow

TensorFlow Recommenders is a library for building recommender system models using TensorFlow.

created at June 26, 2020, 9:38 p.m.

49 +0

1,772 +4

265 -1

decord by dmlc

An efficient video loader for deep learning with smart shuffling that's super easy to digest

created at Jan. 18, 2019, 11:07 p.m.

30 +0

1,690 +12

146 +2

MAgent2 by Farama-Foundation

An engine for high performance multi-agent environments with very large numbers of agents, along with a set of reference environments

created at Sept. 5, 2022, 4:29 p.m.

2 +0

189 +3

32 +0

LAVIS by salesforce

LAVIS - A One-stop Library for Language-Vision Intelligence

created at Aug. 24, 2022, 2:36 a.m.

95 +0

9,028 +46

893 +3

imitation by HumanCompatibleAI

Clean PyTorch implementations of imitation and reward learning algorithms

created at Dec. 8, 2018, 5:15 a.m.

18 +0

1,176 +12

228 +1

pytorch-lightning by PyTorchLightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

created at March 31, 2019, 12:45 a.m.

246 +1

27,234 +50

3,268 +3

hyperlearn by unslothai

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

created at Aug. 27, 2018, 4 p.m.

89 +2

1,651 +13

120 -1

mxnet by ROCm

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

created at April 14, 2017, 6:46 p.m.

30 +0

28 +0

9 +0