Update README.md

README.md

@@ -3,7 +3,83 @@ tags:
 - model_hub_mixin
 - pytorch_model_hub_mixin
 ---
+# AION-Base: Astronomical Omnimodal Network
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
-- Library: [More Information Needed]
-- Docs: [More Information Needed]
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![GitHub](https://img.shields.io/badge/GitHub-PolymathicAI%2FAION-blue)](https://github.com/PolymathicAI/AIO
+N)
+[![Paper](https://img.shields.io/badge/Paper-arXiv-red)](https://arxiv.org/abs/XXXX.XXXXX)
+
+**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 aion
+```
+
+For development:
+```bash
+pip install aion[torch,dev]
+```
+
+## 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
+
+## Model Variants
+
+| Variant | Parameters | Encoder/Decoder | Model Dim |
+|---------|------------|-----------------|-----------|
+| Base    | 300M       | 12/12 blocks    | 768       |
+| Large   | 800M       | 24/24 blocks    | 1024      |
+| XLarge  | 3B         | 24/24 blocks    | 2048      |
+
+## License
+
+MIT License - see https://github.com/PolymathicAI/AION/blob/main/LICENSE for details.
+
+## Citation
+
+@article{aion2024,
+  title={AION: Astronomical Omnimodal Network},
+  author={...},
+  journal={arXiv preprint arXiv:XXXX.XXXXX},
+  year={2025}
+}
+
+---
+Built with ❤ for the astronomical community by https://polymathic-ai.org/