README.md
pyproject.toml
setup.py
jax_unirep/__init__.py
jax_unirep/activations.py
jax_unirep/errors.py
jax_unirep/evotuning.py
jax_unirep/featurize.py
jax_unirep/layers.py
jax_unirep/losses.py
jax_unirep/optimizers.py
jax_unirep/params.py
jax_unirep/sampler.py
jax_unirep/utils.py
jax_unirep.egg-info/PKG-INFO
jax_unirep.egg-info/SOURCES.txt
jax_unirep.egg-info/dependency_links.txt
jax_unirep.egg-info/requires.txt
jax_unirep.egg-info/top_level.txt