Skip to content

Commit

Permalink
Create _neuralgcm (#49)
Browse files Browse the repository at this point in the history 
* Create _neuralgcm

Add the neural GCM as an option to the dabench models. Provide necessary interfaces.

* code pair

* Update _neuralgcm.py

* Update _neuralgcm.py

moved core forecast step to its own method

* Update to store Neural GCM model as an attribute

* Explicitly disable JAX doube precision, getting dtype errors in convs

* Working neural gcm forecast() method with flat statevec inputs

* Removing previous forecast() methods from neuralgcm model class

---------

Co-authored-by: Kylen Solvik <[email protected]>
  • Loading branch information
@StevePny @kysolvik
StevePny and kysolvik authored Oct 2, 2024
1 parent ccd797a commit 0715b0e
Show file tree
Hide file tree
Showing 3 changed files with 1,310 additions and 0 deletions.
Loading

0 comments on commit 0715b0e

Please sign in to comment.