.gitignore
.pre-commit-config.yaml
.python-version
ACKNOWLEDGEMENTS.md
AGENTS.md
CHANGELOG.md
CLA.md
CLAUDE.md
LICENSE.txt
README.md
pyproject.toml
uv.lock
.github/CODEOWNERS
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/cla.yml
.github/workflows/doc-delete-on-pr-closed.yml
.github/workflows/docs.yml
.github/workflows/doctest.yml
.github/workflows/post-coverage.yml
.github/workflows/pr-validation.yml
.github/workflows/pre-commit.yml
.github/workflows/publish-pypi.yml
.github/workflows/release.yml
.github/workflows/run-tests.yml
ci/rules/copyright.py
docs/Makefile
docs/README.md
docs/make.bat
docs/source/conf.py
docs/source/index.rst
docs/source/_ext/source_view.py
docs/source/_static/Emmi_AI_logo_black.svg
docs/source/_static/Emmi_AI_logo_black_on_white.svg
docs/source/_static/Emmi_AI_logo_green.svg
docs/source/_static/Emmi_AI_logo_white.svg
docs/source/_static/Emmi_AI_logo_white_on_black.svg
docs/source/_static/Emmi_AI_logomark_black.svg
docs/source/_static/Emmi_AI_logomark_black_on_white.svg
docs/source/_static/Emmi_AI_logomark_white.svg
docs/source/_static/Emmi_AI_logomark_white_on_black.svg
docs/source/_static/default_furo.scss
docs/source/_static/emmi_docs_theme.css
docs/source/_static/force_light_theme.js
docs/source/_static/noether-framework-logo.svg
docs/source/_static/noether_architecture.png
docs/source/_static/robots.txt
docs/source/_static/shapenet_car_models_data_flow.png
docs/source/_static/heat_transfer/temperature_prediction.png
docs/source/_static/logos_cards/eth.png
docs/source/_static/logos_cards/jku.png
docs/source/_static/logos_cards/sorbonne.png
docs/source/_static/logos_cards/tum.png
docs/source/_static/logos_cards/upenn.png
docs/source/_static/logos_cards/uw.png
docs/source/_static/showcase/750k_points_inference_drivaerml_sample_0000.png
docs/source/_static/showcase/inference_scaling.png
docs/source/_static/showcase/surface_losses.png
docs/source/_static/showcase/volume_losses.png
docs/source/_templates/autoapi/python/package.rst
docs/source/guides/index.rst
docs/source/guides/linux_cuda_setup.rst
docs/source/guides/working_with_cli.rst
docs/source/guides/data/how_to_make_a_custom_dataset.rst
docs/source/guides/data/how_to_use_private_data_source.rst
docs/source/guides/data/how_to_write_a_multistage_pipeline.rst
docs/source/guides/data/how_to_write_a_sample_processor.rst
docs/source/guides/inference/how_to_run_evaluation_on_trained_models.rst
docs/source/guides/training/development_cli.rst
docs/source/guides/training/experiment_tracking.rst
docs/source/guides/training/how_to_initialize.rst
docs/source/guides/training/implement_a_custom_model.rst
docs/source/guides/training/implement_a_custom_trainer.rst
docs/source/guides/training/launch_job.rst
docs/source/guides/training/use_callbacks.rst
docs/source/noether/aero_cfd_python.rst
docs/source/noether/dataset_zoo.rst
docs/source/noether/design_principles_and_limitations.rst
docs/source/noether/heat_transfer.rst
docs/source/noether/index.rst
docs/source/noether/introduction_to_noether_framework.rst
docs/source/noether/key_concepts.rst
docs/source/noether/model_zoo.rst
docs/source/noether/recipe_zoo.rst
docs/source/noether/temperature_prediction.png
docs/source/noether/understanding_the_data_pipeline.rst
docs/source/noether/io/caching.rst
docs/source/reference/config_inheritance.rst
docs/source/reference/hardware_setup.rst
docs/source/reference/prerequisites.rst
docs/source/tutorials/getting_started_install_and_verify.rst
docs/source/tutorials/index.rst
docs/source/tutorials/scaffolding_a_new_project.rst
docs/source/tutorials/training_first_model_with_code.rst
docs/source/tutorials/training_first_model_with_configs.rst
docs/source/tutorials/walkthrough/callbacks.rst
docs/source/tutorials/walkthrough/configuration.rst
docs/source/tutorials/walkthrough/dataset.rst
docs/source/tutorials/walkthrough/index.rst
docs/source/tutorials/walkthrough/introduction.rst
docs/source/tutorials/walkthrough/models.rst
docs/source/tutorials/walkthrough/pipeline.rst
docs/source/tutorials/walkthrough/running_experiments.rst
docs/source/tutorials/walkthrough/showcase.rst
docs/source/tutorials/walkthrough/trainer.rst
recipes/.gitignore
recipes/aero_cfd/README.MD
recipes/aero_cfd/__init__.py
recipes/aero_cfd/pyproject.toml
recipes/aero_cfd/callbacks/__init__.py
recipes/aero_cfd/callbacks/aero_metrics.py
recipes/aero_cfd/callbacks/query_inference.py
recipes/aero_cfd/configs/__init__.py
recipes/aero_cfd/configs/train_ahmedml.yaml
recipes/aero_cfd/configs/train_caeml.yaml
recipes/aero_cfd/configs/train_drivaerml.yaml
recipes/aero_cfd/configs/train_drivaernet.yaml
recipes/aero_cfd/configs/train_shapenet.yaml
recipes/aero_cfd/configs/train_wing.yaml
recipes/aero_cfd/configs/callbacks/training_callbacks_caeml.yaml
recipes/aero_cfd/configs/callbacks/training_callbacks_shapenet.yaml
recipes/aero_cfd/configs/data_specs/caeml.yaml
recipes/aero_cfd/configs/data_specs/emmi_wing.yaml
recipes/aero_cfd/configs/data_specs/shapenet_car.yaml
recipes/aero_cfd/configs/dataset_normalizers/caeml_dataset_normalizers.yaml
recipes/aero_cfd/configs/dataset_normalizers/shapenet_dataset_normalizers.yaml
recipes/aero_cfd/configs/dataset_normalizers/wing_dataset_normalizers.yaml
recipes/aero_cfd/configs/datasets/caeml_dataset.yaml
recipes/aero_cfd/configs/datasets/shapenet_dataset.yaml
recipes/aero_cfd/configs/experiment/ahmedml/ab_upt.yaml
recipes/aero_cfd/configs/experiment/ahmedml/transformer.yaml
recipes/aero_cfd/configs/experiment/ahmedml/transolver.yaml
recipes/aero_cfd/configs/experiment/ahmedml/upt.yaml
recipes/aero_cfd/configs/experiment/drivaerml/ab_upt.yaml
recipes/aero_cfd/configs/experiment/drivaerml/ab_upt_muon_lion.yaml
recipes/aero_cfd/configs/experiment/drivaerml/transformer.yaml
recipes/aero_cfd/configs/experiment/drivaerml/transolver.yaml
recipes/aero_cfd/configs/experiment/drivaerml/upt.yaml
recipes/aero_cfd/configs/experiment/emmi_wing/ab_upt.yaml
recipes/aero_cfd/configs/experiment/shapenet/ab_upt.yaml
recipes/aero_cfd/configs/experiment/shapenet/transformer.yaml
recipes/aero_cfd/configs/experiment/shapenet/transolver.yaml
recipes/aero_cfd/configs/experiment/shapenet/transolver_plusplus.yaml
recipes/aero_cfd/configs/experiment/shapenet/upt.yaml
recipes/aero_cfd/configs/experiment/shapenet/composite/transformer.yaml
recipes/aero_cfd/configs/model/ab_upt.yaml
recipes/aero_cfd/configs/model/composite_transformer.yaml
recipes/aero_cfd/configs/model/transformer.yaml
recipes/aero_cfd/configs/model/transolver.yaml
recipes/aero_cfd/configs/model/upt.yaml
recipes/aero_cfd/configs/optimizer/adamw.yaml
recipes/aero_cfd/configs/optimizer/adamw_transolver.yaml
recipes/aero_cfd/configs/optimizer/lion.yaml
recipes/aero_cfd/configs/optimizer/muon.yaml
recipes/aero_cfd/configs/pipeline/caeml_pipeline.yaml
recipes/aero_cfd/configs/pipeline/shapenet_pipeline.yaml
recipes/aero_cfd/configs/slurm/slurm_config.yaml
recipes/aero_cfd/configs/tracker/development_tracker.yaml
recipes/aero_cfd/configs/tracker/disabled.yaml
recipes/aero_cfd/configs/trainer/caeml_trainer.yaml
recipes/aero_cfd/configs/trainer/shapenet_trainer.yaml
recipes/aero_cfd/images/shapenet_car_models_data_flow.png
recipes/aero_cfd/jobs/train_ahmedml.job
recipes/aero_cfd/jobs/train_drivaerml.job
recipes/aero_cfd/jobs/train_drivaernet.job
recipes/aero_cfd/jobs/train_shapenet.job
recipes/aero_cfd/jobs/experiments/ahmedml_experiments.txt
recipes/aero_cfd/jobs/experiments/drivaerml_experiments.txt
recipes/aero_cfd/jobs/experiments/shapenet_experiments.txt
recipes/aero_cfd/model/__init__.py
recipes/aero_cfd/model/composite_transformer.py
recipes/aero_cfd/model/composite_components/__init__.py
recipes/aero_cfd/model/composite_components/composite_transformer_block.py
recipes/aero_cfd/pipeline/__init__.py
recipes/aero_cfd/pipeline/multistage_pipelines/__init__.py
recipes/aero_cfd/pipeline/multistage_pipelines/aero_multistage.py
recipes/aero_cfd/pipeline/sample_processors/__init__.py
recipes/aero_cfd/pipeline/sample_processors/anchor_point_sampling.py
recipes/aero_cfd/presets/__init__.py
recipes/aero_cfd/presets/ahmedml.py
recipes/aero_cfd/presets/base.py
recipes/aero_cfd/presets/drivaerml.py
recipes/aero_cfd/presets/drivaernet.py
recipes/aero_cfd/presets/emmi_wing.py
recipes/aero_cfd/presets/shapenet_car.py
recipes/aero_cfd/scripts/__init__.py
recipes/aero_cfd/scripts/train_ahmedml.py
recipes/aero_cfd/scripts/train_drivaerml.py
recipes/aero_cfd/scripts/train_drivaernet.py
recipes/aero_cfd/scripts/train_emmi_wing.py
recipes/aero_cfd/scripts/train_shapenet_car.py
recipes/aero_cfd/showcase/README.md
recipes/aero_cfd/showcase/__init__.py
recipes/aero_cfd/showcase/cli.py
recipes/aero_cfd/showcase/model_configs.py
recipes/aero_cfd/showcase/utils/__init__.py
recipes/aero_cfd/showcase/utils/forces.py
recipes/aero_cfd/showcase/utils/vtk_export.py
recipes/aero_cfd/trainers/__init__.py
recipes/aero_cfd/trainers/aerodynamics_cfd.py
recipes/aero_cfd/utils/__init__.py
recipes/aero_cfd/utils/drag_lift.py
recipes/development/README.MD
recipes/development/development_schema.py
recipes/development/callbacks/__init__.py
recipes/development/callbacks/development_callback.py
recipes/development/configs/development_config.yaml
recipes/development/dataset/__init__.py
recipes/development/dataset/development_dataset.py
recipes/development/model/__init__.py
recipes/development/model/development_model.py
recipes/development/pipeline/__init__.py
recipes/development/pipeline/development_pipeline.py
recipes/development/trainer/__init__.py
recipes/development/trainer/development_trainer.py
recipes/heat_transfer/README.md
recipes/heat_transfer/__init__.py
recipes/heat_transfer/pyproject.toml
recipes/heat_transfer/temperature_prediction.png
recipes/heat_transfer/uv.lock
recipes/heat_transfer/callbacks/__init__.py
recipes/heat_transfer/callbacks/heat_transfer_metrics.py
recipes/heat_transfer/callbacks/heat_transfer_visualization.py
recipes/heat_transfer/configs/train_simshift_heatsink.yaml
recipes/heat_transfer/configs/experiment/simshift_heatsink/ab_upt.yaml
recipes/heat_transfer/configs/experiment/simshift_heatsink/transolver.yaml
recipes/heat_transfer/configs/model/ab_upt.yaml
recipes/heat_transfer/configs/model/transformer.yaml
recipes/heat_transfer/configs/slurm/slurm_config.yaml
recipes/heat_transfer/configs/tracker/development_tracker.yaml
recipes/heat_transfer/configs/tracker/disabled.yaml
recipes/heat_transfer/model/__init__.py
recipes/heat_transfer/model/heat_transfer_transformer.py
recipes/heat_transfer/pipeline/__init__.py
recipes/heat_transfer/pipeline/heat_transfer_pipeline.py
src/emmiai_noether.egg-info/PKG-INFO
src/emmiai_noether.egg-info/SOURCES.txt
src/emmiai_noether.egg-info/dependency_links.txt
src/emmiai_noether.egg-info/entry_points.txt
src/emmiai_noether.egg-info/requires.txt
src/emmiai_noether.egg-info/top_level.txt
src/noether/__init__.py
src/noether/core/__init__.py
src/noether/core/constants.py
src/noether/core/types.py
src/noether/core/callbacks/__init__.py
src/noether/core/callbacks/base.py
src/noether/core/callbacks/periodic.py
src/noether/core/callbacks/checkpoint/__init__.py
src/noether/core/callbacks/checkpoint/best_checkpoint.py
src/noether/core/callbacks/checkpoint/checkpoint.py
src/noether/core/callbacks/checkpoint/ema.py
src/noether/core/callbacks/default/__init__.py
src/noether/core/callbacks/default/dataset_stats.py
src/noether/core/callbacks/default/eta.py
src/noether/core/callbacks/default/lr.py
src/noether/core/callbacks/default/online_loss.py
src/noether/core/callbacks/default/param_count.py
src/noether/core/callbacks/default/peak_memory.py
src/noether/core/callbacks/default/progress.py
src/noether/core/callbacks/default/train_time.py
src/noether/core/callbacks/early_stoppers/__init__.py
src/noether/core/callbacks/early_stoppers/base.py
src/noether/core/callbacks/early_stoppers/fixed.py
src/noether/core/callbacks/early_stoppers/metric.py
src/noether/core/callbacks/online/__init__.py
src/noether/core/callbacks/online/best_metric.py
src/noether/core/callbacks/online/track_outputs.py
src/noether/core/configs/__init__.py
src/noether/core/configs/hyperparameters.py
src/noether/core/distributed/__init__.py
src/noether/core/distributed/config.py
src/noether/core/distributed/gather.py
src/noether/core/distributed/reduce.py
src/noether/core/distributed/utils.py
src/noether/core/distributed/run/__init__.py
src/noether/core/distributed/run/managed.py
src/noether/core/distributed/run/unmanaged.py
src/noether/core/extractors/__init__.py
src/noether/core/extractors/forward_hook.py
src/noether/core/factory/__init__.py
src/noether/core/factory/base.py
src/noether/core/factory/dataset.py
src/noether/core/factory/optimizer.py
src/noether/core/factory/schedule.py
src/noether/core/factory/utils.py
src/noether/core/initializers/__init__.py
src/noether/core/initializers/base.py
src/noether/core/initializers/checkpoint.py
src/noether/core/initializers/previous_run.py
src/noether/core/initializers/resume.py
src/noether/core/models/__init__.py
src/noether/core/models/base.py
src/noether/core/models/composite.py
src/noether/core/models/model.py
src/noether/core/optimizer/__init__.py
src/noether/core/optimizer/lion.py
src/noether/core/optimizer/muon_composite.py
src/noether/core/optimizer/optimizer_wrapper.py
src/noether/core/optimizer/param_group_modifiers/__init__.py
src/noether/core/optimizer/param_group_modifiers/base.py
src/noether/core/optimizer/param_group_modifiers/lr_scale_by_name.py
src/noether/core/optimizer/param_group_modifiers/weight_decay_by_name.py
src/noether/core/presets/__init__.py
src/noether/core/presets/base.py
src/noether/core/presets/model_defaults.py
src/noether/core/providers/__init__.py
src/noether/core/providers/metric_property.py
src/noether/core/providers/path.py
src/noether/core/schedules/__init__.py
src/noether/core/schedules/base.py
src/noether/core/schedules/constant.py
src/noether/core/schedules/cosine.py
src/noether/core/schedules/custom.py
src/noether/core/schedules/functional.py
src/noether/core/schedules/linear.py
src/noether/core/schedules/linear_warmup_cosine_decay.py
src/noether/core/schedules/polynomial.py
src/noether/core/schedules/step.py
src/noether/core/schemas/__init__.py
src/noether/core/schemas/callbacks.py
src/noether/core/schemas/dataset.py
src/noether/core/schemas/filemap.py
src/noether/core/schemas/initializers.py
src/noether/core/schemas/lib.py
src/noether/core/schemas/mixins.py
src/noether/core/schemas/normalizers.py
src/noether/core/schemas/optimizers.py
src/noether/core/schemas/schedules.py
src/noether/core/schemas/schema.py
src/noether/core/schemas/slurm.py
src/noether/core/schemas/statistics.py
src/noether/core/schemas/trackers.py
src/noether/core/schemas/trainers.py
src/noether/core/schemas/models/__init__.py
src/noether/core/schemas/models/ab_upt.py
src/noether/core/schemas/models/base.py
src/noether/core/schemas/models/transformer.py
src/noether/core/schemas/models/transolver.py
src/noether/core/schemas/models/upt.py
src/noether/core/schemas/modules/__init__.py
src/noether/core/schemas/modules/attention.py
src/noether/core/schemas/modules/blocks.py
src/noether/core/schemas/modules/decoders.py
src/noether/core/schemas/modules/encoders.py
src/noether/core/schemas/modules/mlp.py
src/noether/core/schemas/modules/untied.py
src/noether/core/schemas/modules/layers/__init__.py
src/noether/core/schemas/modules/layers/continuous_sincos_embedding.py
src/noether/core/schemas/modules/layers/drop_path.py
src/noether/core/schemas/modules/layers/layer_scale.py
src/noether/core/schemas/modules/layers/linear_projection.py
src/noether/core/schemas/modules/layers/rope_frequency.py
src/noether/core/schemas/modules/layers/scalar_conditioner.py
src/noether/core/scripts/__init__.py
src/noether/core/scripts/validate_pytorch_compatibilities.py
src/noether/core/trackers/__init__.py
src/noether/core/trackers/base.py
src/noether/core/trackers/noop.py
src/noether/core/trackers/tensorboard.py
src/noether/core/trackers/trackio_tracker.py
src/noether/core/trackers/wandb_tracker.py
src/noether/core/utils/__init__.py
src/noether/core/utils/bidict.py
src/noether/core/utils/code.py
src/noether/core/utils/model.py
src/noether/core/utils/seed.py
src/noether/core/utils/common/__init__.py
src/noether/core/utils/common/naming.py
src/noether/core/utils/common/path.py
src/noether/core/utils/common/stopwatch.py
src/noether/core/utils/common/typing.py
src/noether/core/utils/common/validation.py
src/noether/core/utils/logging/__init__.py
src/noether/core/utils/logging/base.py
src/noether/core/utils/logging/formatting.py
src/noether/core/utils/logging/progress.py
src/noether/core/utils/platform/__init__.py
src/noether/core/utils/platform/system.py
src/noether/core/utils/platform/worker.py
src/noether/core/utils/torch/__init__.py
src/noether/core/utils/torch/amp.py
src/noether/core/utils/torch/device.py
src/noether/core/utils/training/__init__.py
src/noether/core/utils/training/counter.py
src/noether/core/utils/training/schedule_wrapper.py
src/noether/core/utils/training/training_iteration.py
src/noether/core/writers/__init__.py
src/noether/core/writers/checkpoint_writer.py
src/noether/core/writers/log_writer.py
src/noether/core/writers/prefixed_log_writer.py
src/noether/data/__init__.py
src/noether/data/container.py
src/noether/data/stats.py
src/noether/data/base/__init__.py
src/noether/data/base/dataset.py
src/noether/data/base/subset.py
src/noether/data/base/wrapper.py
src/noether/data/base/wrappers/__init__.py
src/noether/data/base/wrappers/property_subset.py
src/noether/data/base/wrappers/repeat.py
src/noether/data/base/wrappers/shuffle.py
src/noether/data/base/wrappers/subset.py
src/noether/data/base/wrappers/timing.py
src/noether/data/datasets/README.md
src/noether/data/datasets/__init__.py
src/noether/data/datasets/cfd/__init__.py
src/noether/data/datasets/cfd/dataset.py
src/noether/data/datasets/cfd/caeml/README.MD
src/noether/data/datasets/cfd/caeml/__init__.py
src/noether/data/datasets/cfd/caeml/dataset.py
src/noether/data/datasets/cfd/caeml/filemap.py
src/noether/data/datasets/cfd/caeml/preprocessing.py
src/noether/data/datasets/cfd/caeml/ahmedml/__init__.py
src/noether/data/datasets/cfd/caeml/ahmedml/dataset.py
src/noether/data/datasets/cfd/caeml/ahmedml/split.py
src/noether/data/datasets/cfd/caeml/ahmedml/stats.yaml
src/noether/data/datasets/cfd/caeml/drivaerml/__init__.py
src/noether/data/datasets/cfd/caeml/drivaerml/dataset.py
src/noether/data/datasets/cfd/caeml/drivaerml/split.py
src/noether/data/datasets/cfd/caeml/drivaerml/stats.yaml
src/noether/data/datasets/cfd/drivaernet/README.MD
src/noether/data/datasets/cfd/drivaernet/__init__.py
src/noether/data/datasets/cfd/drivaernet/dataset.py
src/noether/data/datasets/cfd/drivaernet/preprocessing.py
src/noether/data/datasets/cfd/drivaernet/stats.yaml
src/noether/data/datasets/cfd/emmi_wing/README.MD
src/noether/data/datasets/cfd/emmi_wing/__init__.py
src/noether/data/datasets/cfd/emmi_wing/dataset.py
src/noether/data/datasets/cfd/emmi_wing/dataset_hf.py
src/noether/data/datasets/cfd/emmi_wing/split.py
src/noether/data/datasets/cfd/emmi_wing/split_hf.py
src/noether/data/datasets/cfd/emmi_wing/stats.yaml
src/noether/data/datasets/cfd/shapenet_car/README.MD
src/noether/data/datasets/cfd/shapenet_car/__init__.py
src/noether/data/datasets/cfd/shapenet_car/dataset.py
src/noether/data/datasets/cfd/shapenet_car/filemap.py
src/noether/data/datasets/cfd/shapenet_car/preprocessing.py
src/noether/data/datasets/cfd/shapenet_car/split.py
src/noether/data/datasets/cfd/shapenet_car/stats.yaml
src/noether/data/datasets/cfd/simshift_heatsink/__init__.py
src/noether/data/datasets/cfd/simshift_heatsink/config.py
src/noether/data/datasets/cfd/simshift_heatsink/dataset.py
src/noether/data/datasets/cfd/simshift_heatsink/stats.json
src/noether/data/pipeline/__init__.py
src/noether/data/pipeline/batch_processor.py
src/noether/data/pipeline/collator.py
src/noether/data/pipeline/multistage.py
src/noether/data/pipeline/sample_processor.py
src/noether/data/pipeline/batch_processors/__init__.py
src/noether/data/pipeline/batch_processors/moment_normalization.py
src/noether/data/pipeline/batch_processors/position_normalization.py
src/noether/data/pipeline/batch_processors/rename_keys.py
src/noether/data/pipeline/collators/__init__.py
src/noether/data/pipeline/collators/concat_sparse_tensor.py
src/noether/data/pipeline/collators/default.py
src/noether/data/pipeline/collators/field_decoder.py
src/noether/data/pipeline/collators/sparse_tensor_offset.py
src/noether/data/pipeline/sample_processors/__init__.py
src/noether/data/pipeline/sample_processors/concat_tensor.py
src/noether/data/pipeline/sample_processors/default_tensor.py
src/noether/data/pipeline/sample_processors/drop_outliers.py
src/noether/data/pipeline/sample_processors/duplicate_keys.py
src/noether/data/pipeline/sample_processors/moment_normalization.py
src/noether/data/pipeline/sample_processors/point_sampling.py
src/noether/data/pipeline/sample_processors/position_normalization.py
src/noether/data/pipeline/sample_processors/rename_keys.py
src/noether/data/pipeline/sample_processors/replace_key.py
src/noether/data/pipeline/sample_processors/supernode_sampling.py
src/noether/data/preprocessors/__init__.py
src/noether/data/preprocessors/base.py
src/noether/data/preprocessors/compose.py
src/noether/data/preprocessors/normalizers.py
src/noether/data/preprocessors/numpy_to_torch_tensor.py
src/noether/data/preprocessors/types.py
src/noether/data/preprocessors/utils.py
src/noether/data/samplers/__init__.py
src/noether/data/samplers/interleaved_sampler.py
src/noether/data/samplers/internals.py
src/noether/data/tools/__init__.py
src/noether/data/tools/calculate_statistics.py
src/noether/inference/__init__.py
src/noether/inference/cli/__init__.py
src/noether/inference/cli/main_inference.py
src/noether/inference/runners/__init__.py
src/noether/inference/runners/inference_runner.py
src/noether/io/__init__.py
src/noether/io/aistore.py
src/noether/io/credentials.py
src/noether/io/logging_config.py
src/noether/io/logging_progress.py
src/noether/io/providers.py
src/noether/io/verification.py
src/noether/io/checkpoint/__init__.py
src/noether/io/checkpoint/iohash.py
src/noether/io/checkpoint/resolver.py
src/noether/io/checkpoint/types.py
src/noether/io/checkpoint/providers/__init__.py
src/noether/io/checkpoint/providers/file_provider.py
src/noether/io/checkpoint/providers/huggingface_provider.py
src/noether/io/checkpoint/providers/s3_provider.py
src/noether/io/cli/__init__.py
src/noether/io/cli/cli.py
src/noether/io/cli/cli_aws.py
src/noether/io/cli/cli_checkpoint.py
src/noether/io/cli/cli_huggingface.py
src/noether/io/cli/cli_utils.py
src/noether/io/cli/cli_verification.py
src/noether/io/diskcache/__init__.py
src/noether/io/diskcache/cache_benchmark.py
src/noether/io/diskcache/lru_cache.py
src/noether/io/interfaces/__init__.py
src/noether/io/interfaces/azure.py
src/noether/io/interfaces/gcp.py
src/noether/io/interfaces/huggingface.py
src/noether/io/interfaces/s3.py
src/noether/io/interfaces/s3_uploader.py
src/noether/modeling/__init__.py
src/noether/modeling/functional/__init__.py
src/noether/modeling/functional/geometric.py
src/noether/modeling/functional/init.py
src/noether/modeling/functional/logscale.py
src/noether/modeling/functional/modulation.py
src/noether/modeling/functional/rope.py
src/noether/modeling/models/README.MD
src/noether/modeling/models/__init__.py
src/noether/modeling/models/ab_upt.py
src/noether/modeling/models/aerodynamics.py
src/noether/modeling/models/transformer.py
src/noether/modeling/models/transolver.py
src/noether/modeling/models/upt.py
src/noether/modeling/modules/__init__.py
src/noether/modeling/modules/activations.py
src/noether/modeling/modules/untied.py
src/noether/modeling/modules/attention/__init__.py
src/noether/modeling/modules/attention/dot_product.py
src/noether/modeling/modules/attention/perceiver.py
src/noether/modeling/modules/attention/transolver.py
src/noether/modeling/modules/attention/transolver_plusplus.py
src/noether/modeling/modules/attention/anchor_attention/__init__.py
src/noether/modeling/modules/attention/anchor_attention/cross.py
src/noether/modeling/modules/attention/anchor_attention/joint.py
src/noether/modeling/modules/attention/anchor_attention/mixed.py
src/noether/modeling/modules/attention/anchor_attention/multi_branch.py
src/noether/modeling/modules/attention/anchor_attention/self_anchor.py
src/noether/modeling/modules/blocks/__init__.py
src/noether/modeling/modules/blocks/perceiver.py
src/noether/modeling/modules/blocks/transformer.py
src/noether/modeling/modules/decoders/__init__.py
src/noether/modeling/modules/decoders/deep_perceiver.py
src/noether/modeling/modules/encoders/__init__.py
src/noether/modeling/modules/encoders/supernode_pooling.py
src/noether/modeling/modules/layers/__init__.py
src/noether/modeling/modules/layers/continuous_sincos_embed.py
src/noether/modeling/modules/layers/drop_path.py
src/noether/modeling/modules/layers/layer_scale.py
src/noether/modeling/modules/layers/linear_projection.py
src/noether/modeling/modules/layers/rope_frequency.py
src/noether/modeling/modules/layers/scalar_conditioner.py
src/noether/modeling/modules/layers/transformer_batchnorm.py
src/noether/modeling/modules/mlp/__init__.py
src/noether/modeling/modules/mlp/mlp.py
src/noether/modeling/modules/mlp/upactdown_mlp.py
src/noether/scaffold/__init__.py
src/noether/scaffold/choices.py
src/noether/scaffold/cli.py
src/noether/scaffold/config.py
src/noether/scaffold/file_manager.py
src/noether/scaffold/template_files/__init__.py
src/noether/scaffold/template_files/pyproject.toml
src/noether/scaffold/template_files/callbacks/__init__.py
src/noether/scaffold/template_files/callbacks/base.py
src/noether/scaffold/template_files/configs/base_experiment.yaml
src/noether/scaffold/template_files/configs/inference_config.yaml
src/noether/scaffold/template_files/configs/tracker/disabled.yaml
src/noether/scaffold/template_files/configs/tracker/tensorboard.yaml
src/noether/scaffold/template_files/configs/tracker/trackio.yaml
src/noether/scaffold/template_files/configs/tracker/wandb.yaml
src/noether/scaffold/template_files/datasets/__init__.py
src/noether/scaffold/template_files/datasets/base.py
src/noether/scaffold/template_files/models/__init__.py
src/noether/scaffold/template_files/models/base.py
src/noether/scaffold/template_files/pipelines/__init__.py
src/noether/scaffold/template_files/pipelines/collators/__init__.py
src/noether/scaffold/template_files/pipelines/postprocessors/__init__.py
src/noether/scaffold/template_files/pipelines/preprocessors/__init__.py
src/noether/scaffold/template_files/schemas/collator/base_collator_config.py
src/noether/scaffold/template_files/schemas/datasets/base_dataset_config.py
src/noether/scaffold/template_files/schemas/models/base_model_config.py
src/noether/scaffold/template_files/trainer/__init__.py
src/noether/scaffold/template_files/trainer/base.py
src/noether/training/__init__.py
src/noether/training/callbacks/__init__.py
src/noether/training/callbacks/offline_loss.py
src/noether/training/callbacks/profiler.py
src/noether/training/cli/__init__.py
src/noether/training/cli/development.py
src/noether/training/cli/main_train.py
src/noether/training/cli/submit_job.py
src/noether/training/runners/__init__.py
src/noether/training/runners/hydra_runner.py
src/noether/training/trainers/__init__.py
src/noether/training/trainers/base.py
src/noether/training/trainers/types.py
src/noether/training/trainers/weighted_loss.py
tests/__init__.py
tests/conftest.py
tests/integration/__init__.py
tests/integration/noether/data/test_dataset_wrappers.py
tests/integration/noether/data/test_pipeline_determinism.py
tests/integration/noether/modeling/conftest.py
tests/integration/noether/modeling/test_model_integration.py
tests/integration/noether/modeling/test_model_outputs.py
tests/integration/noether/training/test_gradient_accumulation.py
tests/integration/noether/training/test_skip_nan_loss.py
tests/test_files/__init__.py
tests/test_files/test_run_id/__init__.py
tests/test_files/test_run_id/train/__init__.py
tests/test_files/test_run_id/train/checkpoints/__init__.py
tests/test_files/test_run_id/train/checkpoints/dummy_model_ema_cp=latest_model.th
tests/test_training_pipeline/__init__.py
tests/test_training_pipeline/conftest.py
tests/test_training_pipeline/test_training_pipeline.py
tests/test_training_pipeline/dummy_project/__init__.py
tests/test_training_pipeline/dummy_project/callbacks/__init__.py
tests/test_training_pipeline/dummy_project/callbacks/base_callback.py
tests/test_training_pipeline/dummy_project/configs/__init__.py
tests/test_training_pipeline/dummy_project/configs/base_experiment.yaml
tests/test_training_pipeline/dummy_project/configs/callbacks/__init__.py
tests/test_training_pipeline/dummy_project/configs/callbacks/base_callback.yaml
tests/test_training_pipeline/dummy_project/configs/datasets/__init__.py
tests/test_training_pipeline/dummy_project/configs/datasets/base_dataset.yaml
tests/test_training_pipeline/dummy_project/configs/model/__init__.py
tests/test_training_pipeline/dummy_project/configs/model/base_model.yaml
tests/test_training_pipeline/dummy_project/configs/optim/__init__.py
tests/test_training_pipeline/dummy_project/configs/optim/adamw.yaml
tests/test_training_pipeline/dummy_project/configs/trainer/__init__.py
tests/test_training_pipeline/dummy_project/configs/trainer/base_trainer.yaml
tests/test_training_pipeline/dummy_project/datasets/__init__.py
tests/test_training_pipeline/dummy_project/datasets/base_dataset.py
tests/test_training_pipeline/dummy_project/models/__init__.py
tests/test_training_pipeline/dummy_project/models/base_model.py
tests/test_training_pipeline/dummy_project/pipelines/__init__.py
tests/test_training_pipeline/dummy_project/pipelines/base_multistage_pipeline.py
tests/test_training_pipeline/dummy_project/pipelines/collators/__init__.py
tests/test_training_pipeline/dummy_project/pipelines/postprocessors/__init__.py
tests/test_training_pipeline/dummy_project/pipelines/preprocessors/__init__.py
tests/test_training_pipeline/dummy_project/schemas/__init__.py
tests/test_training_pipeline/dummy_project/schemas/config_schema.py
tests/test_training_pipeline/dummy_project/schemas/callbacks/__init__.py
tests/test_training_pipeline/dummy_project/schemas/callbacks/base_callback_config.py
tests/test_training_pipeline/dummy_project/schemas/collator/__init__.py
tests/test_training_pipeline/dummy_project/schemas/collator/base_collator_config.py
tests/test_training_pipeline/dummy_project/schemas/datasets/__init__.py
tests/test_training_pipeline/dummy_project/schemas/datasets/base_dataset_config.py
tests/test_training_pipeline/dummy_project/schemas/models/__init__.py
tests/test_training_pipeline/dummy_project/schemas/models/base_model_config.py
tests/test_training_pipeline/dummy_project/schemas/trainer/__init__.py
tests/test_training_pipeline/dummy_project/schemas/trainer/base_trainer_config.py
tests/test_training_pipeline/dummy_project/trainer/__init__.py
tests/test_training_pipeline/dummy_project/trainer/base_trainer.py
tests/unit/__init__.py
tests/unit/noether/__init__.py
tests/unit/noether/test_distributed.py
tests/unit/noether/test_distributed_gather_nondistributed.py
tests/unit/noether/test_distributed_run_parse_devices.py
tests/unit/noether/core/__init__.py
tests/unit/noether/core/callbacks/__init__.py
tests/unit/noether/core/callbacks/test_base.py
tests/unit/noether/core/callbacks/test_metric_early_stopper.py
tests/unit/noether/core/callbacks/test_online.py
tests/unit/noether/core/callbacks/test_periodic.py
tests/unit/noether/core/callbacks/checkpoint/__init__.py
tests/unit/noether/core/callbacks/checkpoint/test_best_checkpoint.py
tests/unit/noether/core/callbacks/checkpoint/test_checkpoint.py
tests/unit/noether/core/callbacks/checkpoint/test_ema.py
tests/unit/noether/core/callbacks/default/__init__.py
tests/unit/noether/core/callbacks/default/test_eta.py
tests/unit/noether/core/callbacks/default/test_param_count.py
tests/unit/noether/core/callbacks/default/test_progress.py
tests/unit/noether/core/configs/__init__.py
tests/unit/noether/core/configs/test_hyperparameters.py
tests/unit/noether/core/distributed/__init__.py
tests/unit/noether/core/distributed/test_config.py
tests/unit/noether/core/distributed/test_managed_run.py
tests/unit/noether/core/distributed/test_unmanaged_run.py
tests/unit/noether/core/extractors/__init__.py
tests/unit/noether/core/extractors/test_forward_hook.py
tests/unit/noether/core/factory/__init__.py
tests/unit/noether/core/factory/test_base.py
tests/unit/noether/core/factory/test_dataset.py
tests/unit/noether/core/factory/test_optimizer.py
tests/unit/noether/core/factory/test_schedule.py
tests/unit/noether/core/factory/test_utils.py
tests/unit/noether/core/initializers/__init__.py
tests/unit/noether/core/initializers/test_checkpoint.py
tests/unit/noether/core/initializers/test_previous_run.py
tests/unit/noether/core/initializers/test_resume.py
tests/unit/noether/core/models/__init__.py
tests/unit/noether/core/models/test_base.py
tests/unit/noether/core/models/test_composite.py
tests/unit/noether/core/models/test_single.py
tests/unit/noether/core/optimizer/__init__.py
tests/unit/noether/core/optimizer/test_muon_composite.py
tests/unit/noether/core/optimizer/test_optimizer_wrapper.py
tests/unit/noether/core/optimizer/param_group_modifiers/__init__.py
tests/unit/noether/core/optimizer/param_group_modifiers/test_lr_scale_by_name.py
tests/unit/noether/core/optimizer/param_group_modifiers/test_weight_decay_by_name.py
tests/unit/noether/core/presets/__init__.py
tests/unit/noether/core/presets/test_base.py
tests/unit/noether/core/providers/__init__.py
tests/unit/noether/core/providers/test_metric_property.py
tests/unit/noether/core/providers/test_path.py
tests/unit/noether/core/schedules/__init__.py
tests/unit/noether/core/schedules/test_constant.py
tests/unit/noether/core/schedules/test_cosine.py
tests/unit/noether/core/schedules/test_custom_schedule.py
tests/unit/noether/core/schedules/test_decreasing_progress.py
tests/unit/noether/core/schedules/test_linear.py
tests/unit/noether/core/schedules/test_linear_warmup_cosine_decay.py
tests/unit/noether/core/schedules/test_sequential_percent.py
tests/unit/noether/core/schedules/test_sequential_step.py
tests/unit/noether/core/schedules/test_step.py
tests/unit/noether/core/schemas/__init__.py
tests/unit/noether/core/schemas/test_filemap.py
tests/unit/noether/core/schemas/test_lib.py
tests/unit/noether/core/schemas/test_normalizers.py
tests/unit/noether/core/schemas/test_optimizers.py
tests/unit/noether/core/schemas/test_statistics.py
tests/unit/noether/core/schemas/models/conftest.py
tests/unit/noether/core/schemas/models/test_ab_upt.py
tests/unit/noether/core/schemas/models/test_upt.py
tests/unit/noether/core/trackers/__init__.py
tests/unit/noether/core/trainer/test_trainer_callbacks.py
tests/unit/noether/core/trainer/test_trainer_checkpoint.py
tests/unit/noether/core/utils/__init__.py
tests/unit/noether/core/utils/test_bidict.py
tests/unit/noether/core/utils/test_code.py
tests/unit/noether/core/utils/test_model.py
tests/unit/noether/core/utils/test_seed.py
tests/unit/noether/core/utils/common/__init__.py
tests/unit/noether/core/utils/common/test_naming.py
tests/unit/noether/core/utils/common/test_path.py
tests/unit/noether/core/utils/common/test_stopwatch.py
tests/unit/noether/core/utils/common/test_validation.py
tests/unit/noether/core/utils/logging/__init__.py
tests/unit/noether/core/utils/logging/test_base.py
tests/unit/noether/core/utils/logging/test_formatting.py
tests/unit/noether/core/utils/platform/__init__.py
tests/unit/noether/core/utils/platform/test_system.py
tests/unit/noether/core/utils/platform/test_worker.py
tests/unit/noether/core/utils/torch/__init__.py
tests/unit/noether/core/utils/torch/test_amp.py
tests/unit/noether/core/utils/torch/test_device.py
tests/unit/noether/core/utils/training/__init__.py
tests/unit/noether/core/utils/training/test_counter.py
tests/unit/noether/core/utils/training/test_schedule_wrapper.py
tests/unit/noether/core/utils/training/test_training_iteration.py
tests/unit/noether/core/writers/__init__.py
tests/unit/noether/core/writers/test_checkpoint_writer.py
tests/unit/noether/core/writers/test_log_writer.py
tests/unit/noether/data/__init__.py
tests/unit/noether/data/test_container.py
tests/unit/noether/data/test_stats.py
tests/unit/noether/data/base/__init__.py
tests/unit/noether/data/base/test_base.py
tests/unit/noether/data/base/wrappers/__init__.py
tests/unit/noether/data/base/wrappers/test_property_subset.py
tests/unit/noether/data/base/wrappers/test_repeat.py
tests/unit/noether/data/base/wrappers/test_shuffle.py
tests/unit/noether/data/base/wrappers/test_subset.py
tests/unit/noether/data/datasets/__init__.py
tests/unit/noether/data/datasets/cfd/__init__.py
tests/unit/noether/data/datasets/cfd/caeml/__init__.py
tests/unit/noether/data/datasets/cfd/caeml/test_ahmedml_dataset.py
tests/unit/noether/data/datasets/cfd/simshift_heatsink/__init__.py
tests/unit/noether/data/datasets/cfd/simshift_heatsink/test_simshift_heatsink_dataset.py
tests/unit/noether/data/pipeline/__init__.py
tests/unit/noether/data/pipeline/test_default.py
tests/unit/noether/data/pipeline/test_drop_outliers.py
tests/unit/noether/data/pipeline/test_duplicate_keys.py
tests/unit/noether/data/pipeline/test_moment_normalization.py
tests/unit/noether/data/pipeline/test_multistage.py
tests/unit/noether/data/pipeline/test_point_sampling.py
tests/unit/noether/data/pipeline/test_position_normalization.py
tests/unit/noether/data/pipeline/test_rename_keys.py
tests/unit/noether/data/pipeline/test_replace_key.py
tests/unit/noether/data/pipeline/test_supernode_sampling.py
tests/unit/noether/data/pipeline/collators/__init__.py
tests/unit/noether/data/pipeline/collators/test_concat_sparse_tensor.py
tests/unit/noether/data/pipeline/collators/test_default.py
tests/unit/noether/data/pipeline/collators/test_field_decoder.py
tests/unit/noether/data/preprocessors/__init__.py
tests/unit/noether/data/preprocessors/test_base.py
tests/unit/noether/data/preprocessors/test_compose.py
tests/unit/noether/data/preprocessors/test_normalizers.py
tests/unit/noether/data/preprocessors/test_numpy_to_torch_tensor.py
tests/unit/noether/data/preprocessors/test_utils.py
tests/unit/noether/data/samplers/__init__.py
tests/unit/noether/data/samplers/test_interleaved_sampler.py
tests/unit/noether/data/samplers/test_internals.py
tests/unit/noether/inference/__init__.py
tests/unit/noether/inference/test_cli.py
tests/unit/noether/inference/test_runner.py
tests/unit/noether/io/__init__.py
tests/unit/noether/io/test_cli_aws.py
tests/unit/noether/io/test_cli_huggingface.py
tests/unit/noether/io/test_credentials.py
tests/unit/noether/io/test_verification.py
tests/unit/noether/io/checkpoint/__init__.py
tests/unit/noether/io/checkpoint/test_huggingface_provider.py
tests/unit/noether/io/diskcache/__init__.py
tests/unit/noether/io/diskcache/test_lru_cache.py
tests/unit/noether/io/interfaces/__init__.py
tests/unit/noether/io/interfaces/test_huggingface.py
tests/unit/noether/io/interfaces/test_s3.py
tests/unit/noether/modeling/__init__.py
tests/unit/noether/modeling/test_functional.py
tests/unit/noether/modeling/test_geometric.py
tests/unit/noether/modeling/test_geometric_performance.py
tests/unit/noether/modeling/test_geometric_triton.py
tests/unit/noether/modeling/models/__init__.py
tests/unit/noether/modeling/models/test_ab_upt.py
tests/unit/noether/modeling/models/test_ab_upt_performance.py
tests/unit/noether/modeling/models/test_transolver.py
tests/unit/noether/modeling/modules/__init__.py
tests/unit/noether/modeling/modules/test_activations.py
tests/unit/noether/modeling/modules/test_rope.py
tests/unit/noether/modeling/modules/test_untied.py
tests/unit/noether/modeling/modules/test_untied_performance.py
tests/unit/noether/modeling/modules/attention/__init__.py
tests/unit/noether/modeling/modules/attention/expected_outputs.py
tests/unit/noether/modeling/modules/attention/test_dot_product.py
tests/unit/noether/modeling/modules/attention/test_joint.py
tests/unit/noether/modeling/modules/attention/test_mixed.py
tests/unit/noether/modeling/modules/attention/test_perceiver.py
tests/unit/noether/modeling/modules/attention/test_transolver.py
tests/unit/noether/modeling/modules/attention/test_transolverplusplus.py
tests/unit/noether/modeling/modules/blocks/__init__.py
tests/unit/noether/modeling/modules/blocks/expected_output.py
tests/unit/noether/modeling/modules/blocks/test_irregular_nat.py
tests/unit/noether/modeling/modules/blocks/test_perceiver.py
tests/unit/noether/modeling/modules/blocks/test_transformer.py
tests/unit/noether/modeling/modules/decoders/__init__.py
tests/unit/noether/modeling/modules/decoders/expected_output.py
tests/unit/noether/modeling/modules/decoders/test_deep_perceiver.py
tests/unit/noether/modeling/modules/encoders/__init__.py
tests/unit/noether/modeling/modules/encoders/expected_output.py
tests/unit/noether/modeling/modules/encoders/test_supernode_pooling.py
tests/unit/noether/modeling/modules/encoders/test_supernode_pooling_posonly.py
tests/unit/noether/modeling/modules/layers/__init__.py
tests/unit/noether/modeling/modules/layers/expected_output.py
tests/unit/noether/modeling/modules/layers/test_continuous_sincos_embed.py
tests/unit/noether/modeling/modules/layers/test_drop_path.py
tests/unit/noether/modeling/modules/layers/test_layer_scale.py
tests/unit/noether/modeling/modules/layers/test_linear_projection.py
tests/unit/noether/modeling/modules/layers/test_rope_frequency.py
tests/unit/noether/modeling/modules/layers/test_scalar_conditioner.py
tests/unit/noether/modeling/modules/layers/test_transformer_batchnorm.py
tests/unit/noether/modeling/modules/mlp/__init__.py
tests/unit/noether/modeling/modules/mlp/expected_output.py
tests/unit/noether/modeling/modules/mlp/test_upactdown_mlp.py
tests/unit/noether/scaffold/__init__.py
tests/unit/noether/scaffold/test_cli.py
tests/unit/noether/scaffold/test_config.py
tests/unit/noether/scaffold/test_generator.py
tests/unit/noether/training/__init__.py
tests/unit/noether/training/test_base_trainer.py
tests/unit/noether/training/test_hydra_runner.py
tests/unit/noether/training/test_offline_loss.py
tests/unit/noether/training/test_periodic_iterator.py
tests/unit/noether/training/cli/test_submit_job.py