.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CLAUDE.md
LICENSE
README.md
pyproject.toml
.github/workflows/deploy-doc.yml
.github/workflows/publish-pypi.yml
.github/workflows/test-doc.yml
.github/workflows/test.yaml
aion/__init__.py
aion/modalities.py
aion/model.py
aion/codecs/__init__.py
aion/codecs/base.py
aion/codecs/catalog.py
aion/codecs/config.py
aion/codecs/image.py
aion/codecs/manager.py
aion/codecs/scalar.py
aion/codecs/scalar_field.py
aion/codecs/spectrum.py
aion/codecs/utils.py
aion/codecs/modules/__init__.py
aion/codecs/modules/convblocks.py
aion/codecs/modules/convnext.py
aion/codecs/modules/ema.py
aion/codecs/modules/magvit.py
aion/codecs/modules/spectrum.py
aion/codecs/modules/subsampler.py
aion/codecs/modules/utils.py
aion/codecs/preprocessing/band_to_index.py
aion/codecs/preprocessing/image.py
aion/codecs/quantizers/__init__.py
aion/codecs/quantizers/lookup_free_quantization.py
aion/codecs/quantizers/scalar.py
aion/fourm/__init__.py
aion/fourm/decoder_embeddings.py
aion/fourm/encoder_embeddings.py
aion/fourm/fm.py
aion/fourm/fm_utils.py
aion/fourm/generate.py
aion/fourm/generation_utils.py
aion/fourm/lora_utils.py
aion/fourm/modality_info.py
aion/fourm/modality_transforms.py
aion/fourm/text_utils.py
assets/aion.png
docs/Makefile
docs/api.md
docs/architecture.md
docs/conf.py
docs/index.md
docs/installation.md
docs/usage.md
docs/_static/aion.png
docs/_static/data_mixture.png
docs/_static/embeddings.png
docs/_static/image.png
docs/_static/polymathic_logo.png
docs/_static/style.css
notebooks/StellarTutorial.ipynb
notebooks/Tutorial.ipynb
polymathic_aion.egg-info/PKG-INFO
polymathic_aion.egg-info/SOURCES.txt
polymathic_aion.egg-info/dependency_links.txt
polymathic_aion.egg-info/requires.txt
polymathic_aion.egg-info/top_level.txt
scripts/export_gaia_codecs.py
scripts/export_hsc_codecs.py
tests/conftest.py
tests/codecs/test_catalog_codec.py
tests/codecs/test_codec_manager.py
tests/codecs/test_image_codec.py
tests/codecs/test_load_codecs.py
tests/codecs/test_scalar_codec.py
tests/codecs/test_scalar_field_codec.py
tests/codecs/test_spectrum_codec.py
tests/test_data/EBV_codec_decoded_batch.pt
tests/test_data/EBV_codec_encoded_batch.pt
tests/test_data/EBV_codec_input_batch.pt
tests/test_data/FLUX_G_codec_decoded_batch.pt
tests/test_data/FLUX_G_codec_encoded_batch.pt
tests/test_data/FLUX_G_codec_input_batch.pt
tests/test_data/FLUX_I_codec_decoded_batch.pt
tests/test_data/FLUX_I_codec_encoded_batch.pt
tests/test_data/FLUX_I_codec_input_batch.pt
tests/test_data/FLUX_R_codec_decoded_batch.pt
tests/test_data/FLUX_R_codec_encoded_batch.pt
tests/test_data/FLUX_R_codec_input_batch.pt
tests/test_data/FLUX_W1_codec_decoded_batch.pt
tests/test_data/FLUX_W1_codec_encoded_batch.pt
tests/test_data/FLUX_W1_codec_input_batch.pt
tests/test_data/FLUX_W2_codec_decoded_batch.pt
tests/test_data/FLUX_W2_codec_encoded_batch.pt
tests/test_data/FLUX_W2_codec_input_batch.pt
tests/test_data/FLUX_W3_codec_decoded_batch.pt
tests/test_data/FLUX_W3_codec_encoded_batch.pt
tests/test_data/FLUX_W3_codec_input_batch.pt
tests/test_data/FLUX_W4_codec_decoded_batch.pt
tests/test_data/FLUX_W4_codec_encoded_batch.pt
tests/test_data/FLUX_W4_codec_input_batch.pt
tests/test_data/FLUX_Z_codec_decoded_batch.pt
tests/test_data/FLUX_Z_codec_encoded_batch.pt
tests/test_data/FLUX_Z_codec_input_batch.pt
tests/test_data/SHAPE_E1_codec_decoded_batch.pt
tests/test_data/SHAPE_E1_codec_encoded_batch.pt
tests/test_data/SHAPE_E1_codec_input_batch.pt
tests/test_data/SHAPE_E2_codec_decoded_batch.pt
tests/test_data/SHAPE_E2_codec_encoded_batch.pt
tests/test_data/SHAPE_E2_codec_input_batch.pt
tests/test_data/SHAPE_R_codec_decoded_batch.pt
tests/test_data/SHAPE_R_codec_encoded_batch.pt
tests/test_data/SHAPE_R_codec_input_batch.pt
tests/test_data/SPECTRUM_decoded_batch.pt
tests/test_data/SPECTRUM_encoded_batch.pt
tests/test_data/SPECTRUM_input_batch.pt
tests/test_data/Z_codec_decoded_batch.pt
tests/test_data/Z_codec_encoded_batch.pt
tests/test_data/Z_codec_input_batch.pt
tests/test_data/a_g_codec_decoded_batch.pt
tests/test_data/a_g_codec_encoded_batch.pt
tests/test_data/a_g_codec_input_batch.pt
tests/test_data/a_i_codec_decoded_batch.pt
tests/test_data/a_i_codec_encoded_batch.pt
tests/test_data/a_i_codec_input_batch.pt
tests/test_data/a_r_codec_decoded_batch.pt
tests/test_data/a_r_codec_encoded_batch.pt
tests/test_data/a_r_codec_input_batch.pt
tests/test_data/a_y_codec_decoded_batch.pt
tests/test_data/a_y_codec_encoded_batch.pt
tests/test_data/a_y_codec_input_batch.pt
tests/test_data/a_z_codec_decoded_batch.pt
tests/test_data/a_z_codec_encoded_batch.pt
tests/test_data/a_z_codec_input_batch.pt
tests/test_data/bp_coefficients_codec_decoded_batch.pt
tests/test_data/bp_coefficients_codec_encoded_batch.pt
tests/test_data/bp_coefficients_codec_input_batch.pt
tests/test_data/catalog_codec_decoded_batch.pt
tests/test_data/catalog_codec_encoded_batch.pt
tests/test_data/catalog_codec_input_batch.pt
tests/test_data/dec_codec_decoded_batch.pt
tests/test_data/dec_codec_encoded_batch.pt
tests/test_data/dec_codec_input_batch.pt
tests/test_data/g_cmodel_mag_codec_decoded_batch.pt
tests/test_data/g_cmodel_mag_codec_encoded_batch.pt
tests/test_data/g_cmodel_mag_codec_input_batch.pt
tests/test_data/i_cmodel_mag_codec_decoded_batch.pt
tests/test_data/i_cmodel_mag_codec_encoded_batch.pt
tests/test_data/i_cmodel_mag_codec_input_batch.pt
tests/test_data/i_sdssshape_shape11_codec_decoded_batch.pt
tests/test_data/i_sdssshape_shape11_codec_encoded_batch.pt
tests/test_data/i_sdssshape_shape11_codec_input_batch.pt
tests/test_data/i_sdssshape_shape12_codec_decoded_batch.pt
tests/test_data/i_sdssshape_shape12_codec_encoded_batch.pt
tests/test_data/i_sdssshape_shape12_codec_input_batch.pt
tests/test_data/i_sdssshape_shape22_codec_decoded_batch.pt
tests/test_data/i_sdssshape_shape22_codec_encoded_batch.pt
tests/test_data/i_sdssshape_shape22_codec_input_batch.pt
tests/test_data/image_codec_decoded_batch.pt
tests/test_data/image_codec_encoded_batch.pt
tests/test_data/image_codec_input_batch.pt
tests/test_data/parallax_codec_decoded_batch.pt
tests/test_data/parallax_codec_encoded_batch.pt
tests/test_data/parallax_codec_input_batch.pt
tests/test_data/phot_bp_mean_flux_codec_decoded_batch.pt
tests/test_data/phot_bp_mean_flux_codec_encoded_batch.pt
tests/test_data/phot_bp_mean_flux_codec_input_batch.pt
tests/test_data/phot_g_mean_flux_codec_decoded_batch.pt
tests/test_data/phot_g_mean_flux_codec_encoded_batch.pt
tests/test_data/phot_g_mean_flux_codec_input_batch.pt
tests/test_data/phot_rp_mean_flux_codec_decoded_batch.pt
tests/test_data/phot_rp_mean_flux_codec_encoded_batch.pt
tests/test_data/phot_rp_mean_flux_codec_input_batch.pt
tests/test_data/r_cmodel_mag_codec_decoded_batch.pt
tests/test_data/r_cmodel_mag_codec_encoded_batch.pt
tests/test_data/r_cmodel_mag_codec_input_batch.pt
tests/test_data/ra_codec_decoded_batch.pt
tests/test_data/ra_codec_encoded_batch.pt
tests/test_data/ra_codec_input_batch.pt
tests/test_data/rp_coefficients_codec_decoded_batch.pt
tests/test_data/rp_coefficients_codec_encoded_batch.pt
tests/test_data/rp_coefficients_codec_input_batch.pt
tests/test_data/scalar-field_codec_decoded_batch.pt
tests/test_data/scalar-field_codec_encoded_batch.pt
tests/test_data/scalar-field_codec_input_batch.pt
tests/test_data/y_cmodel_mag_codec_decoded_batch.pt
tests/test_data/y_cmodel_mag_codec_encoded_batch.pt
tests/test_data/y_cmodel_mag_codec_input_batch.pt
tests/test_data/z_cmodel_mag_codec_decoded_batch.pt
tests/test_data/z_cmodel_mag_codec_encoded_batch.pt
tests/test_data/z_cmodel_mag_codec_input_batch.pt