pytorch implemention of trajGRU.
PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a...