optax by deepmind

Optax is a gradient processing and optimization library for JAX.

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

36 +0

1,534 +4

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

694 +3

94 +1

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

430 +3

42 +1

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

4,730 +37

549 +1

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,236 +42

1,581 +5

PyTorch-BigGraph by facebookresearch

Generate embeddings from large-scale graph-structured data.

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

90 +0

3,356 +2

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

6,081 +43

703 +8

acme by deepmind

A library of reinforcement learning components and agents

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

83 +0

3,412 +2

419 +1

d3rlpy by takuseno

An offline deep reinforcement learning library

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

28 +0

1,234 +3

226 +1

graph_nets by deepmind

Build Graph Nets in Tensorflow

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

223 +0

5,331 -1

782 +0

ngboost by stanfordmlgroup

Natural Gradient Boosting for Probabilistic Prediction

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

44 +0

1,612 +8

214 -1

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

13,144 +15

2,978 +1

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,072 +1

119 +1

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

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

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

167 +0

keras-cv by keras-team

Industry-strength Computer Vision workflows with Keras

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

29 +0

969 +7

316 +2

keras-nlp by keras-team

Modular Natural Language Processing workflows with Keras

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

27 +0

722 +4

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

105 +0