LICENSE
README.md
pyproject.toml
jax_unirep/__init__.py
jax_unirep/activations.py
jax_unirep/errors.py
jax_unirep/evotuning.py
jax_unirep/featurize.py
jax_unirep/losses.py
jax_unirep/models.py
jax_unirep/sampler.py
jax_unirep/utils.py
jax_unirep/version.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
jax_unirep/weights/uniref50/1900_weights/model_weights.npz
jax_unirep/weights/uniref50/256_weights/model_weights.npz
jax_unirep/weights/uniref50/64_weights/model_weights.npz
tests/test_activations.py
tests/test_evotuning.py
tests/test_evotuning_models.py
tests/test_featurize.py
tests/test_losses.py
tests/test_models.py
tests/test_regression.py
tests/test_sampler.py
tests/test_utils.py