LICENSE
README.md
pyproject.toml
neuralgcm/__init__.py
neuralgcm/demo.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/atmosphere/__init__.py
neuralgcm/experimental/atmosphere/diagnostics.py
neuralgcm/experimental/atmosphere/diagnostics_test.py
neuralgcm/experimental/atmosphere/equations.py
neuralgcm/experimental/atmosphere/observation_operators.py
neuralgcm/experimental/atmosphere/observation_operators_test.py
neuralgcm/experimental/atmosphere/parameterizations.py
neuralgcm/experimental/atmosphere/state_conversion.py
neuralgcm/experimental/atmosphere/transforms.py
neuralgcm/experimental/atmosphere/transforms_test.py
neuralgcm/experimental/core/__init__.py
neuralgcm/experimental/core/api.py
neuralgcm/experimental/core/array_transforms.py
neuralgcm/experimental/core/boundaries.py
neuralgcm/experimental/core/boundaries_test.py
neuralgcm/experimental/core/checkpointing.py
neuralgcm/experimental/core/checkpointing_test.py
neuralgcm/experimental/core/coordinates.py
neuralgcm/experimental/core/coordinates_test.py
neuralgcm/experimental/core/data_specs.py
neuralgcm/experimental/core/diagnostics.py
neuralgcm/experimental/core/diagnostics_test.py
neuralgcm/experimental/core/dynamic_io.py
neuralgcm/experimental/core/equations.py
neuralgcm/experimental/core/feature_transforms.py
neuralgcm/experimental/core/feature_transforms_test.py
neuralgcm/experimental/core/fiddle_tags.py
neuralgcm/experimental/core/field_utils.py
neuralgcm/experimental/core/field_utils_test.py
neuralgcm/experimental/core/interpolators.py
neuralgcm/experimental/core/interpolators_test.py
neuralgcm/experimental/core/learned_transforms.py
neuralgcm/experimental/core/learned_transforms_test.py
neuralgcm/experimental/core/module_utils.py
neuralgcm/experimental/core/nnx_compat.py
neuralgcm/experimental/core/normalizations.py
neuralgcm/experimental/core/normalizations_test.py
neuralgcm/experimental/core/observation_operators.py
neuralgcm/experimental/core/observation_operators_test.py
neuralgcm/experimental/core/orographies.py
neuralgcm/experimental/core/parallelism.py
neuralgcm/experimental/core/parallelism_test.py
neuralgcm/experimental/core/pytree_utils.py
neuralgcm/experimental/core/pytree_utils_test.py
neuralgcm/experimental/core/random_processes.py
neuralgcm/experimental/core/random_processes_test.py
neuralgcm/experimental/core/scales.py
neuralgcm/experimental/core/spatial_filters.py
neuralgcm/experimental/core/spherical_transforms.py
neuralgcm/experimental/core/spherical_transforms_test.py
neuralgcm/experimental/core/standard_layers.py
neuralgcm/experimental/core/standard_layers_test.py
neuralgcm/experimental/core/step_filters.py
neuralgcm/experimental/core/time_integrators.py
neuralgcm/experimental/core/time_integrators_test.py
neuralgcm/experimental/core/towers.py
neuralgcm/experimental/core/towers_test.py
neuralgcm/experimental/core/transformer_layers.py
neuralgcm/experimental/core/transformer_layers_test.py
neuralgcm/experimental/core/transforms.py
neuralgcm/experimental/core/transforms_test.py
neuralgcm/experimental/core/typing.py
neuralgcm/experimental/core/units.py
neuralgcm/experimental/core/xarray_utils.py
neuralgcm/experimental/core/xarray_utils_test.py
neuralgcm/experimental/xreader/__init__.py
neuralgcm/experimental/xreader/iterators.py
neuralgcm/experimental/xreader/iterators_test.py
neuralgcm/experimental/xreader/stencils.py
neuralgcm/experimental/xreader/stencils_test.py
neuralgcm/legacy/__init__.py
neuralgcm/legacy/api.py
neuralgcm/legacy/api_test.py
neuralgcm/legacy/correctors.py
neuralgcm/legacy/decoders.py
neuralgcm/legacy/decoders_test.py
neuralgcm/legacy/diagnostics.py
neuralgcm/legacy/embeddings.py
neuralgcm/legacy/encoders.py
neuralgcm/legacy/encoders_test.py
neuralgcm/legacy/equations.py
neuralgcm/legacy/features.py
neuralgcm/legacy/features_test.py
neuralgcm/legacy/filters.py
neuralgcm/legacy/forcings.py
neuralgcm/legacy/forcings_test.py
neuralgcm/legacy/gin_utils.py
neuralgcm/legacy/initializers.py
neuralgcm/legacy/initializers_test.py
neuralgcm/legacy/integrators.py
neuralgcm/legacy/layers.py
neuralgcm/legacy/layers_test.py
neuralgcm/legacy/mappings.py
neuralgcm/legacy/mappings_test.py
neuralgcm/legacy/model_builder.py
neuralgcm/legacy/model_utils.py
neuralgcm/legacy/model_utils_test.py
neuralgcm/legacy/models_test.py
neuralgcm/legacy/optimization.py
neuralgcm/legacy/optimization_test.py
neuralgcm/legacy/orographies.py
neuralgcm/legacy/parameterizations.py
neuralgcm/legacy/parameterizations_test.py
neuralgcm/legacy/perturbations.py
neuralgcm/legacy/physics_specifications.py
neuralgcm/legacy/steps.py
neuralgcm/legacy/stochastic.py
neuralgcm/legacy/stochastic_test.py
neuralgcm/legacy/towers.py
neuralgcm/legacy/towers_test.py
neuralgcm/legacy/transforms.py
neuralgcm/legacy/transforms_test.py