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,208 +2

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

16 +1

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,092 +2

1,399 -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,076 +4

325 +2

treelite by dmlc

Universal model exchange and serialization format for decision tree forests

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

40 +0

718 +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 +0

817 +1

114 +0

RecBole by RUCAIBox

A unified, comprehensive and efficient recommendation library

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

40 +0

3,244 +14

588 +0

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,433 +6

399 +4

ranking by tensorflow

Learning to Rank in TensorFlow

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

97 +0

2,720 +1

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,447 +5

108 +1

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,777 +5

265 +0

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,695 +5

147 +1

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

191 +2

33 +1

LAVIS by salesforce

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

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

95 +0

9,073 +45

899 +6

imitation by HumanCompatibleAI

Clean PyTorch implementations of imitation and reward learning algorithms

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

18 +0

1,180 +4

229 +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 +0

27,286 +52

3,274 +6

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 +0

1,663 +12

120 +0

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