LICENSE
README.md
pyproject.toml
terrax/__init__.py
terrax/init_test.py
terrax.egg-info/PKG-INFO
terrax.egg-info/SOURCES.txt
terrax.egg-info/dependency_links.txt
terrax.egg-info/requires.txt
terrax.egg-info/top_level.txt
terrax/atmosphere/__init__.py
terrax/atmosphere/diagnostics.py
terrax/atmosphere/diagnostics_test.py
terrax/atmosphere/equations.py
terrax/atmosphere/equations_test.py
terrax/atmosphere/fixers.py
terrax/atmosphere/fixers_test.py
terrax/atmosphere/idealized_states.py
terrax/atmosphere/idealized_states_test.py
terrax/atmosphere/interpolators.py
terrax/atmosphere/interpolators_test.py
terrax/atmosphere/observation_operators.py
terrax/atmosphere/observation_operators_test.py
terrax/atmosphere/parameterizations.py
terrax/atmosphere/state_conversion.py
terrax/atmosphere/transforms.py
terrax/atmosphere/transforms_test.py
terrax/auxiliary/__init__.py
terrax/auxiliary/models.py
terrax/auxiliary/models_test.py
terrax/core/__init__.py
terrax/core/api.py
terrax/core/api_test.py
terrax/core/array_transforms.py
terrax/core/boundaries.py
terrax/core/boundaries_test.py
terrax/core/checkpointing.py
terrax/core/checkpointing_test.py
terrax/core/coordinates.py
terrax/core/coordinates_test.py
terrax/core/data_specs.py
terrax/core/data_specs_test.py
terrax/core/diagnostics.py
terrax/core/diagnostics_test.py
terrax/core/dynamic_io.py
terrax/core/equations.py
terrax/core/feature_transforms.py
terrax/core/feature_transforms_test.py
terrax/core/fiddle_tags.py
terrax/core/field_utils.py
terrax/core/field_utils_test.py
terrax/core/interpolators.py
terrax/core/interpolators_test.py
terrax/core/learned_transforms.py
terrax/core/learned_transforms_test.py
terrax/core/module_utils.py
terrax/core/module_utils_test.py
terrax/core/normalizations.py
terrax/core/normalizations_test.py
terrax/core/observation_operators.py
terrax/core/observation_operators_test.py
terrax/core/orographies.py
terrax/core/parallelism.py
terrax/core/parallelism_test.py
terrax/core/pytree_utils.py
terrax/core/pytree_utils_test.py
terrax/core/random_processes.py
terrax/core/random_processes_test.py
terrax/core/scales.py
terrax/core/scan_utils.py
terrax/core/scan_utils_test.py
terrax/core/spatial_filters.py
terrax/core/spatial_filters_test.py
terrax/core/spherical_harmonics.py
terrax/core/spherical_harmonics_test.py
terrax/core/standard_layers.py
terrax/core/standard_layers_test.py
terrax/core/step_filters.py
terrax/core/step_filters_test.py
terrax/core/time_integrators.py
terrax/core/towers.py
terrax/core/towers_test.py
terrax/core/transformer_layers.py
terrax/core/transformer_layers_test.py
terrax/core/transforms.py
terrax/core/transforms_test.py
terrax/core/typing.py
terrax/core/units.py
terrax/core/xarray_utils.py
terrax/core/xarray_utils_test.py
terrax/couplers/__init__.py
terrax/couplers/generic.py
terrax/couplers/generic_test.py
terrax/inference/__init__.py
terrax/inference/dynamic_inputs.py
terrax/inference/dynamic_inputs_test.py
terrax/inference/runner.py
terrax/inference/runner_test.py
terrax/inference/streaming.py
terrax/inference/streaming_test.py
terrax/inference/timing.py
terrax/inference/timing_test.py
terrax/metrics/__init__.py
terrax/metrics/aggregation.py
terrax/metrics/aggregation_test.py
terrax/metrics/base.py
terrax/metrics/binning.py
terrax/metrics/deterministic_losses.py
terrax/metrics/deterministic_metrics.py
terrax/metrics/deterministic_test.py
terrax/metrics/evaluators.py
terrax/metrics/evaluators_test.py
terrax/metrics/probabilistic_losses.py
terrax/metrics/probabilistic_metrics.py
terrax/metrics/probabilistic_test.py
terrax/metrics/scaling.py
terrax/metrics/scaling_test.py
terrax/metrics/weighting.py
terrax/metrics/weighting_test.py
terrax/toy_model_examples/__init__.py
terrax/toy_model_examples/data_model.py
terrax/toy_model_examples/data_model_test.py
terrax/toy_model_examples/lorenz96.py
terrax/toy_model_examples/lorenz96_test.py
terrax/training/__init__.py
terrax/training/checkpointing.py
terrax/training/data_loading.py
terrax/training/data_loading_test.py
terrax/training/model_calibrators.py
terrax/training/model_calibrators_test.py
terrax/training/train_utils.py
terrax/training/trainer.py
terrax/training/trainer_test.py
terrax/xreader/__init__.py
terrax/xreader/iterators.py
terrax/xreader/iterators_test.py
terrax/xreader/stencils.py
terrax/xreader/stencils_test.py