polymathic-ai
aion-base
[](https://opensource.org/licenses/MIT) [](https://github.com/PolymathicAI/AION) [](https://arxiv.org/abs/2510.17960) [](https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb) AION-base is a 300M parameter large omnimodal model specifically designed for astronomical surveys. It integrates 39 distinct astronomical data types and enables adaptation to a wide range of astronomical tasks through multimodal masked modeling. - Architecture: Encoder-Decoder Transformer (12 blocks each, 768 dim, 12 heads) - Parameters: 300M - Training: Multimodal Masked Modeling (4M) on astronomical survey data - Modalities: 39 data types including imaging, spectra, catalogs, and photometry AION-Base processes data from major astronomical surveys: - Imaging: Legacy Survey, HSC Wide - Spectra: SDSS, DESI - Catalog: Legacy Survey entries - Gaia: BP/RP spectra, parallax, coordinates, photometry - Photometry: Legacy Survey (g,r,i,z + WISE), HSC (g,r,i,z,y) - Shape: Ellipticity and morphological parameters - GitHub: https://github.com/PolymathicAI/AION - Tutorial: https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb MIT License - see https://github.com/PolymathicAI/AION/blob/main/LICENSE for details. --- Built with ❤ for the astronomical community by https://polymathic-ai.org/