PyPI page
Home page
Author:
Daniil Tiapkin, Arten Agarkov, Nikita Morozov, Ian Maksimov, Timofei Gritsaev, Sergey Samsonov
Summary:
Environments for Generative Flow Networks in JAX
Latest version:
0.0.1
Required dependencies:
chex
|
flashbax
|
jax
|
jax-tqdm
|
jaxlib
|
jaxtyping
|
matplotlib
|
networkx
|
numpy
|
omegaconf
Optional dependencies:
equinox
|
flashbax
|
hydra-core
|
jax-tqdm
|
jraph
|
optax
|
orbax-checkpoint
|
pillow
|
pytest
|
wandb
Downloads last day:
0
Downloads last week:
31
Downloads last month:
238