optax by deepmind

Optax is a gradient processing and optimization library for JAX.

created at June 12, 2020, 3:45 p.m.

36 +1

1,530 +12

161 +2

OpenFE by IIIS-Li-Group

OpenFE: automated feature generation with expert-level performance

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

8 +0

691 +4

93 +2

skrl by Toni-SM

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

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

10 +0

427 +4

41 +0

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.

35 +1

4,693 +53

548 +5

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.

61 +0

8,194 +48

1,576 -1

PyTorch-BigGraph by facebookresearch

Generate embeddings from large-scale graph-structured data.

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

90 +0

3,354 +0

445 +0

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.

40 +1

6,038 +49

695 +1

acme by deepmind

A library of reinforcement learning components and agents

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

83 +0

3,410 +10

418 +0

d3rlpy by takuseno

An offline deep reinforcement learning library

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

28 +0

1,231 +5

225 +0

graph_nets by deepmind

Build Graph Nets in Tensorflow

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

223 -1

5,332 +2

782 +1

ngboost by stanfordmlgroup

Natural Gradient Boosting for Probabilistic Prediction

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

44 +0

1,604 +3

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

169 +1

13,129 +25

2,977 +6

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.

28 +0

1,071 -1

118 +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,032 -1

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

126 +5

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

38 +0

1,296 +2

167 +0

keras-cv by keras-team

Industry-strength Computer Vision workflows with Keras

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

29 -1

962 +2

314 +2

keras-nlp by keras-team

Modular Natural Language Processing workflows with Keras

created at May 28, 2020, 11:03 p.m.

27 +0

718 +5

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

25 +0

650 +1

105 +0