Hugging Face's logo

polymathic-ai
/
aion-base

Safetensors
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
Community
1
aion-base
File size: 2,227 Bytes 
aace99b
 
 
 
57270a3
 
 
 
 
 
 
aace99b
3cb8240
aace99b
423e97e
cab0446
3731704
 
423e97e
3cb8240
423e97e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
57270a3
423e97e
 
 
 
 
 
 
 
 
 
 
 
 
57270a3
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
 
---
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
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)
[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-blue?logo=github)](https://github.com/PolymathicAI/AION)
[![arXiv](https://img.shields.io/badge/arXiv-2510.17960-b31b1b.svg)](https://arxiv.org/abs/2510.17960)
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](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.

## 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
- 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/