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/experimental/__init__.py
neuralgcm/experimental/api.py
neuralgcm/experimental/checkpointing.py
neuralgcm/experimental/circulation_models.py
neuralgcm/experimental/coordinates.py
neuralgcm/experimental/coordinates_test.py
neuralgcm/experimental/data_specs.py
neuralgcm/experimental/diagnostics.py
neuralgcm/experimental/dynamic_io.py
neuralgcm/experimental/equations.py
neuralgcm/experimental/interpolators.py
neuralgcm/experimental/interpolators_test.py
neuralgcm/experimental/module_utils.py
neuralgcm/experimental/orographies.py
neuralgcm/experimental/process_parameterizations.py
neuralgcm/experimental/pytree_mappings.py
neuralgcm/experimental/pytree_mappings_test.py
neuralgcm/experimental/pytree_transforms.py
neuralgcm/experimental/pytree_transforms_test.py
neuralgcm/experimental/pytree_utils.py
neuralgcm/experimental/pytree_utils_test.py
neuralgcm/experimental/random_processes.py
neuralgcm/experimental/random_processes_test.py
neuralgcm/experimental/scales.py
neuralgcm/experimental/spatial_filters.py
neuralgcm/experimental/standard_layers.py
neuralgcm/experimental/standard_layers_test.py
neuralgcm/experimental/state_conversion.py
neuralgcm/experimental/time_integrators.py
neuralgcm/experimental/time_integrators_test.py
neuralgcm/experimental/towers.py
neuralgcm/experimental/towers_test.py
neuralgcm/experimental/typing.py
neuralgcm/experimental/typing_test.py
neuralgcm/experimental/units.py
neuralgcm/experimental/xarray_utils.py
neuralgcm/experimental/coordax/__init__.py
neuralgcm/experimental/coordax/core.py
neuralgcm/experimental/coordax/core_test.py
neuralgcm/experimental/coordax/named_axes.py
neuralgcm/experimental/coordax/named_axes_test.py
neuralgcm/experimental/coordax/testing.py
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