---
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
extra_gated_fields:
  GitHub username: text
  Affiliation: text
  I want to use this model on the following data:
    type: select
    options:
    - Legacy Survey
    - HSC
    - SDSS
    - DESI
    - Gaia
  I want to use this model for the following science case: text
  I agree to provide feedback on the model and API through GitHub issues during the alpha-testing phase: checkbox
  I agree to use this model for non-commercial use ONLY: checkbox
license: mit
datasets:
- MultimodalUniverse/legacysurvey
- MultimodalUniverse/hsc
- MultimodalUniverse/gaia
- MultimodalUniverse/sdss
- MultimodalUniverse/desi
---
# AION-1: Astronomical Omnimodal Network

[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

**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.

## Model Details

- **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

## Quick Start

```python
from aion import AION

# Load the pretrained model
model = AION.from_pretrained('polymathic-ai/aion-base')

# Your astronomical analysis begins here!
```

## Supported Data Types

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

## Installation

```bash
pip install polymathic-aion
```


## Resources

- GitHub: https://github.com/PolymathicAI/AION
- Documentation: https://polymathic-ai.github.io/AION/
- Tutorial: https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb

## License

MIT License - see https://github.com/PolymathicAI/AION/blob/main/LICENSE for details.

---
Built with ❤ for the astronomical community by https://polymathic-ai.org/