kyegomez/Aurora
Implementation of the paper: "Aurora: A Foundation Model of the Atmosphere" in PyTorch
Built on a Swin Transformer UNet 3D architecture, Aurora processes spatiotemporal atmospheric data through hierarchical shifted-window attention mechanisms optimized for weather prediction tasks. The model operates on 5D tensor inputs (batch, depth, height, width, channels) representing volumetric atmospheric fields, enabling efficient multi-scale feature extraction across spatial and temporal dimensions. Installable via pip as `aurora-torch`, it integrates directly into PyTorch workflows for atmospheric modeling and weather forecasting applications.
Stars
22
Forks
3
Language
Python
License
MIT
Category
Last pushed
Mar 09, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/agents/kyegomez/Aurora"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
google/adk-python
An open-source, code-first Python toolkit for building, evaluating, and deploying sophisticated...
google/adk-java
An open-source, code-first Java toolkit for building, evaluating, and deploying sophisticated AI...
google/adk-go
An open-source, code-first Go toolkit for building, evaluating, and deploying sophisticated AI...
edu010101/adk-extra-services
A Python package providing additional service implementations for the Google ADK framework (S3,...
njraladdin/adk-typescript
TypeScript port of Google's Agent Development Kit (ADK): An open-source, code-first toolkit for...