LICENSE
README.md
setup.py
examples/example_cnn.py
examples/example_input_space.py
examples/example_mlp.py
examples/example_mnist_mlp.py
jaxzoo/__init__.py
jaxzoo/base_model.py
jaxzoo/cnn.py
jaxzoo/mlp.py
jaxzoo/neural_components.py
jaxzoo/spaces.py
jaxzoo/utils.py
jaxzoo.egg-info/PKG-INFO
jaxzoo.egg-info/SOURCES.txt
jaxzoo.egg-info/dependency_links.txt
jaxzoo.egg-info/top_level.txt
tests/test_spaces.py