LICENSE
README.md
pyproject.toml
neuralgcm/__init__.py
neuralgcm/api.py
neuralgcm/api_test.py
neuralgcm/correctors.py
neuralgcm/decoders.py
neuralgcm/decoders_test.py
neuralgcm/demo.py
neuralgcm/diagnostics.py
neuralgcm/embeddings.py
neuralgcm/encoders.py
neuralgcm/encoders_test.py
neuralgcm/equations.py
neuralgcm/features.py
neuralgcm/features_test.py
neuralgcm/filters.py
neuralgcm/forcings.py
neuralgcm/forcings_test.py
neuralgcm/gin_utils.py
neuralgcm/initializers.py
neuralgcm/initializers_test.py
neuralgcm/integrators.py
neuralgcm/layers.py
neuralgcm/layers_test.py
neuralgcm/mappings.py
neuralgcm/mappings_test.py
neuralgcm/model_builder.py
neuralgcm/model_utils.py
neuralgcm/model_utils_test.py
neuralgcm/models_test.py
neuralgcm/optimization.py
neuralgcm/optimization_test.py
neuralgcm/orographies.py
neuralgcm/parameterizations.py
neuralgcm/parameterizations_test.py
neuralgcm/perturbations.py
neuralgcm/physics_specifications.py
neuralgcm/steps.py
neuralgcm/stochastic.py
neuralgcm/stochastic_test.py
neuralgcm/towers.py
neuralgcm/towers_test.py
neuralgcm/transforms.py
neuralgcm/transforms_test.py
neuralgcm.egg-info/PKG-INFO
neuralgcm.egg-info/SOURCES.txt
neuralgcm.egg-info/dependency_links.txt
neuralgcm.egg-info/requires.txt
neuralgcm.egg-info/top_level.txt
neuralgcm/data/era5_tl31_19590102T00.nc
neuralgcm/data/tl63_stochastic_mini.pkl
neuralgcm/reference_code/datasets.py
neuralgcm/reference_code/experiment.py
neuralgcm/reference_code/linear_transforms.py
neuralgcm/reference_code/metrics.py
neuralgcm/reference_code/metrics_base.py
neuralgcm/reference_code/metrics_util.py
neuralgcm/reference_code/reader.py
neuralgcm/reference_code/stochastic_losses.py
neuralgcm/reference_code/train_utils.py