krzjoa/awesome-python-data-science

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.

Python

7 +0

119 +0

16 +0

GitHub

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.

Python

4 +0

397 +0

51 +0

GitHub

autokeras by keras-team

AutoML library for deep learning

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

Python

302 +0

9,115 +4

1,396 +1

GitHub

mmengine by open-mmlab

OpenMMLab Foundational Library for Training Deep Learning Models

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

Python

25 +0

1,145 +7

340 +0

GitHub

treelite by dmlc

Universal model exchange and serialization format for decision tree forests

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

C++

40 +0

729 +0

98 +0

GitHub

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.

Python

28 +0

852 +2

118 -1

GitHub

RecBole by RUCAIBox

A unified, comprehensive and efficient recommendation library

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

Python

41 +1

3,356 +8

604 +1

GitHub

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.

Python

18 -1

2,558 +5

408 +2

GitHub

ranking by tensorflow

Learning to Rank in TensorFlow

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

Python

97 +0

2,736 +0

474 +1

GitHub

vizro by mckinsey

Vizro is a toolkit for creating modular data visualization applications.

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

Python

18 +0

2,615 +4

116 +0

GitHub

recommenders by tensorflow

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

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

Python

49 +0

1,824 +4

273 +0

GitHub

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.

C++

29 +0

1,820 +7

158 +0

GitHub

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.

C++

3 +0

217 +0

37 +0

GitHub

LAVIS by salesforce

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

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

Jupyter Notebook

97 +1

9,681 +26

950 +6

GitHub

imitation by HumanCompatibleAI

Clean PyTorch implementations of imitation and reward learning algorithms

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

Python

18 +0

1,274 +3

243 +3

GitHub

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.

Python

247 +0

28,015 +57

3,359 +12

GitHub

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.

Jupyter Notebook

89 +0

1,779 +4

127 +1

GitHub

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.

C++

30 +0

28 +0

9 +0

GitHub

tensorflow-upstream by ROCm

TensorFlow ROCm port

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

C++

51 +0

686 -1

93 +0

GitHub

pymoo by anyoptimization

NSGA2, NSGA3, R-NSGA3, MOEAD, Genetic Algorithms (GA), Differential Evolution (DE), CMAES, PSO

created at Sept. 19, 2017, 9:52 p.m.

Python

32 +1

2,215 +9

381 +0

GitHub