| Name | Version | Summary | date |
| mctx |
0.0.5 |
Monte Carlo tree search in JAX. |
2023-11-24 11:52:32 |
| distrax |
0.1.5 |
Distrax: Probability distributions in JAX. |
2023-11-22 12:53:44 |
| 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 |
| 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 |
| 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 |
| alphafold |
2.0.0 |
An implementation of the inference pipeline of AlphaFold v2.0.This is a completely new model that was entered as AlphaFold2 in CASP14 and published in Nature. |
2021-07-16 14:30:25 |
| bsuite |
0.3.5 |
Core RL Behaviour Suite. A collection of reinforcement learning experiments. |
2021-02-18 11:30:41 |
DeepMind