| Name | Version | Summary | date |
| pgmax |
0.6.1 |
Loopy belief propagation for factor graphs on discrete variables in JAX |
2023-08-11 01:52:38 |
| fancyflags |
1.2 |
A Python library for defining dictionary-valued flags. |
2023-07-04 15:11:34 |
| rlax |
0.1.6 |
A library of reinforcement learning building blocks in JAX. |
2023-06-29 15:03:36 |
| dm-pix |
0.4.1 |
PIX is an image processing library in JAX, for JAX. |
2023-06-05 10:35:57 |
| android-env |
1.2.2 |
AndroidEnv environment and library for training agents. |
2023-04-27 17:23:38 |
| dm-launchpad-nightly |
0.5.2.dev20230323 |
Launchpad is a library that simplifies writing distributed programs and seamlessly launching them on a range of supported platforms. |
2023-03-23 10:21:26 |
| dks |
0.1.2 |
A Python library implementing the DKS/TAT neural network transformation method. |
2023-01-30 23:19:15 |
| jmp |
0.0.4 |
JMP is a Mixed Precision library for JAX. |
2023-01-30 12:47:13 |
| dm-env-rpc |
1.1.4 |
A networking protocol for agent-environment communication. |
2023-01-04 15:08:13 |
| dm-env |
1.6 |
A Python interface for Reinforcement Learning environments. |
2022-12-21 00:25:29 |
| einshape |
1.0 |
DSL-based reshaping library for JAX and other frameworks |
2022-12-19 17:09:34 |
| dm-tree |
0.1.8 |
Tree is a library for working with nested data structures. |
2022-12-18 09:46:55 |
| labmaze |
1.0.6 |
LabMaze: DeepMind Lab's text maze generator. |
2022-12-05 18:42:43 |
| jraph |
0.0.6.dev0 |
Jraph: A library for Graph Neural Networks in Jax |
2022-08-12 15:24:47 |
| bsuite |
0.3.5 |
Core RL Behaviour Suite. A collection of reinforcement learning experiments. |
2021-02-18 11:30:41 |
DeepMind