LICENSE
README.md
pyproject.toml
src/astro_emulators_toolkit/__init__.py
src/astro_emulators_toolkit/_typing.py
src/astro_emulators_toolkit/_version.py
src/astro_emulators_toolkit/emulator.py
src/astro_emulators_toolkit/emulator_runtime.py
src/astro_emulators_toolkit/io_trees.py
src/astro_emulators_toolkit/py.typed
src/astro_emulators_toolkit/resolver.py
src/astro_emulators_toolkit/spec.py
src/astro_emulators_toolkit.egg-info/PKG-INFO
src/astro_emulators_toolkit.egg-info/SOURCES.txt
src/astro_emulators_toolkit.egg-info/dependency_links.txt
src/astro_emulators_toolkit.egg-info/requires.txt
src/astro_emulators_toolkit.egg-info/top_level.txt
src/astro_emulators_toolkit/_vendor/__init__.py
src/astro_emulators_toolkit/_vendor/get_soap_jax.py
src/astro_emulators_toolkit/_vendor/soap_jax/LICENSE
src/astro_emulators_toolkit/_vendor/soap_jax/README.txt
src/astro_emulators_toolkit/_vendor/soap_jax/__init__.py
src/astro_emulators_toolkit/_vendor/soap_jax/soap.py
src/astro_emulators_toolkit/bundle/__init__.py
src/astro_emulators_toolkit/bundle/artifact.py
src/astro_emulators_toolkit/bundle/bundle.py
src/astro_emulators_toolkit/bundle/extras.py
src/astro_emulators_toolkit/bundle/hub.py
src/astro_emulators_toolkit/bundle/integrity.py
src/astro_emulators_toolkit/bundle/metadata.py
src/astro_emulators_toolkit/bundle/minmax_sidecars.py
src/astro_emulators_toolkit/bundle/provenance.py
src/astro_emulators_toolkit/bundle/readme.py
src/astro_emulators_toolkit/bundle/release.py
src/astro_emulators_toolkit/bundle/safetensors_io.py
src/astro_emulators_toolkit/bundle/versions.py
src/astro_emulators_toolkit/config/__init__.py
src/astro_emulators_toolkit/config/io.py
src/astro_emulators_toolkit/config/parsing.py
src/astro_emulators_toolkit/config/schema.py
src/astro_emulators_toolkit/data/__init__.py
src/astro_emulators_toolkit/data/array_dataset.py
src/astro_emulators_toolkit/data/loader.py
src/astro_emulators_toolkit/data/mapped_dataset.py
src/astro_emulators_toolkit/data/npy_table.py
src/astro_emulators_toolkit/data/preprocess.py
src/astro_emulators_toolkit/data/protocols.py
src/astro_emulators_toolkit/data/subset.py
src/astro_emulators_toolkit/data/toy.py
src/astro_emulators_toolkit/data/toy_utils.py
src/astro_emulators_toolkit/experimental/__init__.py
src/astro_emulators_toolkit/experimental/runtime_adapters.py
src/astro_emulators_toolkit/experimental/models/__init__.py
src/astro_emulators_toolkit/experimental/models/explicit_wavelength_mlp.py
src/astro_emulators_toolkit/experimental/models/mlp_2d_regression.py
src/astro_emulators_toolkit/experimental/models/siren.py
src/astro_emulators_toolkit/experimental/tasks/__init__.py
src/astro_emulators_toolkit/experimental/tasks/classification.py
src/astro_emulators_toolkit/inference/__init__.py
src/astro_emulators_toolkit/inference/compose.py
src/astro_emulators_toolkit/models/__init__.py
src/astro_emulators_toolkit/models/activations.py
src/astro_emulators_toolkit/models/cannon.py
src/astro_emulators_toolkit/models/canonical_wrappers.py
src/astro_emulators_toolkit/models/mlp.py
src/astro_emulators_toolkit/models/runtime_adapters.py
src/astro_emulators_toolkit/models/transformer_payne.py
src/astro_emulators_toolkit/models/transformer_payne_batch.py
src/astro_emulators_toolkit/optimizers/__init__.py
src/astro_emulators_toolkit/optimizers/_soap.py
src/astro_emulators_toolkit/presets/__init__.py
src/astro_emulators_toolkit/presets/cannon.py
src/astro_emulators_toolkit/presets/mlp.py
src/astro_emulators_toolkit/presets/transformer_payne.py
src/astro_emulators_toolkit/tasks/__init__.py
src/astro_emulators_toolkit/tasks/common.py
src/astro_emulators_toolkit/tasks/regression.py
src/astro_emulators_toolkit/training/__init__.py
src/astro_emulators_toolkit/training/callbacks.py
src/astro_emulators_toolkit/training/checkpoint.py
src/astro_emulators_toolkit/training/fit_backends.py
src/astro_emulators_toolkit/training/paths.py
src/astro_emulators_toolkit/training/solvers.py
src/astro_emulators_toolkit/training/state.py
src/astro_emulators_toolkit/training/trainer.py
src/astro_emulators_toolkit/utils/__init__.py
src/astro_emulators_toolkit/utils/tree.py