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

128 +2

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

400 +1

51 +0

autokeras by keras-team

AutoML library for deep learning

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

301 +0

9,153 +4

1,402 +0

mmengine by open-mmlab

OpenMMLab Foundational Library for Training Deep Learning Models

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

25 +0

1,179 +4

355 +0

treelite by dmlc

Universal model exchange and serialization format for decision tree forests

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

39 +0

738 +0

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

871 +3

119 +0

RecBole by RUCAIBox

A unified, comprehensive and efficient recommendation library

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

42 +0

3,449 +11

617 +2

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.

18 +0

2,624 +7

414 +1

ranking by tensorflow

Learning to Rank in TensorFlow

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

97 +0

2,743 +1

477 +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,705 +19

142 +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.

50 +1

1,848 +5

278 +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.

29 +0

1,891 +11

161 +0

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.

3 +0

228 +1

38 +0

LAVIS by salesforce

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

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

99 +0

9,930 +31

973 +2

imitation by HumanCompatibleAI

Clean PyTorch implementations of imitation and reward learning algorithms

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

19 +0

1,326 +4

246 -1

pytorch-lightning by PyTorchLightning

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

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

250 +1

28,392 +45

3,384 +1

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.

91 +0

1,842 +5

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

tensorflow-upstream by ROCm

TensorFlow ROCm port

created at April 9, 2018, 9:24 p.m.

50 +0

688 +0

94 +0

datacompy by capitalone

Pandas, Polars, Spark, and Snowpark DataFrame comparison for humans and more!

created at March 23, 2018, 1:16 p.m.

23 +0

485 +1

129 +1