Package search
jraph
- Summary: Jraph: A library for Graph Neural Networks in Jax
- Author: DeepMind
- Homepage: https://github.com/deepmind/jraph
- Source: https://github.com/deepmind/jraph
- Repo profile
- Number of releases: 4
- First release: 0.0.1.dev0 on 2021-08-06T09:31:20
- Latest release: 0.0.6.dev0 on 2022-08-12T15:24:45
| Dependency | Optional |
|---|---|
| jax | false |
| jaxlib | false |
| numpy | false |
| absl-py | true |
| dm-haiku | true |
| dm-tree | true |
| frozendict | true |
| optax | true |
| pandas | true |
| scipy | true |