OpenFE by IIIS-Li-Group

OpenFE: automated feature generation with expert-level performance

created at Nov. 21, 2022, 11:20 a.m.

9 +0

781 +0

99 +0

skrl by Toni-SM

Modular reinforcement learning library (on PyTorch and JAX) with support for NVIDIA Isaac Gym, Omniverse Isaac Gym and Isaac Lab

created at Oct. 18, 2021, 7:53 a.m.

9 +0

560 +9

57 +2

cleanrl by vwxyzjn

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

created at June 7, 2019, 4:31 p.m.

38 +1

5,692 +41

642 +2

stable-baselines3 by DLR-RM

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

created at May 5, 2020, 5:52 a.m.

64 -1

9,142 +53

1,704 +6

PyTorch-BigGraph by facebookresearch

Generate embeddings from large-scale graph-structured data.

created at Oct. 1, 2018, 8:41 p.m.

89 +0

3,383 +1

449 -1

Gymnasium by Farama-Foundation

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

created at Sept. 8, 2022, 1:58 a.m.

42 +0

7,372 +67

827 +14

acme by deepmind

A library of reinforcement learning components and agents

created at May 1, 2020, 9:18 a.m.

84 +0

3,514 +11

426 +1

d3rlpy by takuseno

An offline deep reinforcement learning library

created at May 23, 2020, 3:51 p.m.

27 -1

1,327 +3

244 +1

graph_nets by deepmind

Build Graph Nets in Tensorflow

created at Aug. 31, 2018, 8:19 a.m.

225 +0

5,360 +1

782 +0

ngboost by stanfordmlgroup

Natural Gradient Boosting for Probabilistic Prediction

created at June 21, 2018, 10:22 p.m.

44 +0

1,655 +1

218 +0

dgl by dmlc

Python package built to ease deep learning on graph, on top of existing DL frameworks.

created at April 20, 2018, 2:49 p.m.

175 +0

13,549 +17

3,015 +0

catalyst-rl by catalyst-team

None

created at March 17, 2020, 9:29 a.m.

5 +0

46 +0

7 +0

AutoGL by THUMNLab

An autoML framework & toolkit for machine learning on graphs.

created at Nov. 30, 2020, 2:26 p.m.

30 +0

1,088 +0

119 +0

envpool by sail-sg

C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

created at Oct. 20, 2021, 9:11 p.m.

23 +0

1,094 +2

100 +0

Shimmy by Farama-Foundation

An API conversion tool for popular external reinforcement learning environments

created at Oct. 6, 2022, 9:24 p.m.

7 +0

138 -1

20 +0

gnn by tensorflow

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

created at Aug. 10, 2021, 8:21 p.m.

36 +0

1,363 +4

180 +1

keras-cv by keras-team

Industry-strength Computer Vision workflows with Keras

created at May 18, 2020, 10:39 p.m.

31 +0

1,010 +0

330 +0

decision-forests by tensorflow

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.

created at April 22, 2021, 2:37 p.m.

24 +0

660 +0

110 +0

GreatX by EdisonLeeeee

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

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

9 +0

83 +0

12 +1

rlpyt by astooke

Reinforcement Learning in PyTorch

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

52 +0

2,231 -2

324 +0