LICENSE
README.md
pyproject.toml
src/samudra/__init__.py
src/samudra/backend.py
src/samudra/cli.py
src/samudra/config.py
src/samudra/config_base.py
src/samudra/config_schema.py
src/samudra/constants.py
src/samudra/datasets.py
src/samudra/derived_variables.py
src/samudra/eval.py
src/samudra/py.typed
src/samudra/stepper.py
src/samudra/train.py
src/samudra.egg-info/PKG-INFO
src/samudra.egg-info/SOURCES.txt
src/samudra.egg-info/dependency_links.txt
src/samudra.egg-info/entry_points.txt
src/samudra.egg-info/requires.txt
src/samudra.egg-info/top_level.txt
src/samudra/aggregator/__init__.py
src/samudra/aggregator/loss.py
src/samudra/aggregator/main.py
src/samudra/aggregator/metrics.py
src/samudra/aggregator/plotting.py
src/samudra/aggregator/train.py
src/samudra/aggregator/inference/__init__.py
src/samudra/aggregator/inference/main.py
src/samudra/aggregator/inference/reduced.py
src/samudra/aggregator/inference/time_mean.py
src/samudra/aggregator/validate/__init__.py
src/samudra/aggregator/validate/main.py
src/samudra/aggregator/validate/map.py
src/samudra/aggregator/validate/reduced.py
src/samudra/aggregator/validate/rollout.py
src/samudra/aggregator/validate/snapshot.py
src/samudra/aggregator/validate/sub_aggregator.py
src/samudra/configs/data/llc.yaml
src/samudra/configs/data/obs.yaml
src/samudra/configs/data/om4.yaml
src/samudra/configs/data/om4_demo.yaml
src/samudra/configs/samudra_mini_om4/model.yaml
src/samudra/configs/samudra_mini_om4/train.yaml
src/samudra/configs/samudra_multi_om4/eval.yaml
src/samudra/configs/samudra_multi_om4/model.yaml
src/samudra/configs/samudra_multi_om4/train.yaml
src/samudra/configs/samudra_om4/eval.yaml
src/samudra/configs/samudra_om4/model.yaml
src/samudra/configs/samudra_om4/train.yaml
src/samudra/configs/samudra_om4/viz.yaml
src/samudra/configs/samudra_om4_v1/eval.yaml
src/samudra/configs/samudra_om4_v1/model.yaml
src/samudra/configs/samudra_om4_v1/train.yaml
src/samudra/configs/samudra_om4_v1/viz.yaml
src/samudra/configs/samudra_om4_v2/eval.yaml
src/samudra/configs/samudra_om4_v2/model.yaml
src/samudra/configs/samudra_om4_v2/train.yaml
src/samudra/configs/samudra_om4_v2/viz.yaml
src/samudra/configs/samudra_om4_v2_highres/eval.yaml
src/samudra/configs/samudra_om4_v2_highres/model.yaml
src/samudra/configs/samudra_om4_v2_highres/train.yaml
src/samudra/configs/samudra_om4_v2_highres/viz.yaml
src/samudra/configs/search/example.yaml
src/samudra/configs/search/local.yaml
src/samudra/configs/search/osn-artifacts.yaml
src/samudra/configs/search/torch.yaml
src/samudra/metrics/__init__.py
src/samudra/metrics/__main__.py
src/samudra/metrics/comparisons.py
src/samudra/metrics/kernels.py
src/samudra/metrics/observations.py
src/samudra/metrics/report.py
src/samudra/metrics/run.py
src/samudra/metrics/spectra.py
src/samudra/models/__init__.py
src/samudra/models/base.py
src/samudra/models/samudra.py
src/samudra/models/samudra_mini.py
src/samudra/models/samudra_multi.py
src/samudra/models/modules/__init__.py
src/samudra/models/modules/activations.py
src/samudra/models/modules/augment_input.py
src/samudra/models/modules/blocks.py
src/samudra/models/modules/decoder.py
src/samudra/models/modules/encoder.py
src/samudra/models/modules/perceiver.py
src/samudra/models/modules/unet_backbone.py
src/samudra/search/__init__.py
src/samudra/search/__main__.py
src/samudra/search/artifacts.py
src/samudra/search/config.py
src/samudra/search/report.py
src/samudra/search/state.py
src/samudra/search/successive_halving.py
src/samudra/search/worker.py
src/samudra/search/executors/__init__.py
src/samudra/search/executors/base.py
src/samudra/search/executors/local.py
src/samudra/search/executors/slurm.py
src/samudra/utils/__init__.py
src/samudra/utils/atomic.py
src/samudra/utils/compare.py
src/samudra/utils/ctx.py
src/samudra/utils/data.py
src/samudra/utils/device.py
src/samudra/utils/distributed.py
src/samudra/utils/ema.py
src/samudra/utils/llc.py
src/samudra/utils/location.py
src/samudra/utils/logging.py
src/samudra/utils/loss.py
src/samudra/utils/multiton.py
src/samudra/utils/output.py
src/samudra/utils/profiler.py
src/samudra/utils/rollout_validation.py
src/samudra/utils/samplers.py
src/samudra/utils/schedule.py
src/samudra/utils/train.py
src/samudra/utils/train_progress.py
src/samudra/utils/training_summary.py
src/samudra/utils/wandb.py
src/samudra/utils/writer.py
src/samudra/viz/__init__.py
src/samudra/viz/__main__.py
src/samudra/viz/config.py
src/samudra/viz/core.py
src/samudra/viz/norms.py
src/samudra/viz/observations.py
tests/test_aggregator_metrics.py
tests/test_blocks.py
tests/test_canonical_dataset.py
tests/test_config.py
tests/test_datasets.py
tests/test_decoder.py
tests/test_derived_variables.py
tests/test_encoder.py
tests/test_gradient_detaching.py
tests/test_location.py
tests/test_metrics.py
tests/test_metrics_acceptance.py
tests/test_metrics_spectra.py
tests/test_multiton.py
tests/test_patchembed.py
tests/test_perceiver.py
tests/test_periodic.py
tests/test_positional_channels.py
tests/test_quickstart_config.py
tests/test_rollout_validation.py
tests/test_samplers.py
tests/test_samudra_mini.py
tests/test_schedule.py
tests/test_search.py
tests/test_stepper.py
tests/test_train_progress.py
tests/test_trainer.py
tests/test_training_summary.py
tests/test_utils_data.py
tests/test_validate_aggregator.py
tests/test_viz_config.py
tests/test_viz_observations.py
tests/test_wandb.py
tests/test_writer.py