README.md
pyproject.toml
src/jax_cddd/__init__.py
src/jax_cddd/_download.py
src/jax_cddd/decoding.py
src/jax_cddd/gru.py
src/jax_cddd/inference.py
src/jax_cddd/modules.py
src/jax_cddd/param_names.py
src/jax_cddd/params.py
src/jax_cddd/vocab.py
src/jax_cddd.egg-info/PKG-INFO
src/jax_cddd.egg-info/SOURCES.txt
src/jax_cddd.egg-info/dependency_links.txt
src/jax_cddd.egg-info/requires.txt
src/jax_cddd.egg-info/top_level.txt
src/jax_cddd/data/indices_char.npy
tests/test_against_original_code.py
tests/test_against_tf1_reference.py
tests/test_encoder_decoder_shapes.py
tests/test_gru_cell.py
tests/test_inference_chunking.py
tests/test_params.py
tests/test_roundtrip_smiles.py
tests/test_vocab.py