Axial Attention &amp; MetNet: A Neural Weather Model for Precipitation Forecasting
MetNet is a predictive neural network model for weather prediction. It uses axial attention to capture long-range dependencies.