/predRNN-in-Jax

implementation of predRNN v2 in Jax

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers