.dockerignore
.gitattributes
.gitignore
CITATION.cff
LICENSE
README.md
build_docker.sh
delete_bin.ipynb
dockerfile
generate_citation.py
pyproject.toml
.github/workflows/page.yml
.github/workflows/python-publish.yml
.vscode/settings.json
benchmark/.gitignore
benchmark/bench_inference.py
benchmark/bench_metrics.py
benchmark/bench_train.py
benchmark/test.py
benchmark/utils/__init__.py
benchmark/utils/data.py
benchmark/utils/metrics.py
benchmark/utils/model.py
benchmark/utils/trainer.py
ci_cd/docker_upload.sh
ci_cd/donodo_template.json
dev/.gitignore
dev/AutoregressiveTime.ipynb
dev/Blocks.ipynb
dev/DTW.ipynb
dev/Inference.ipynb
dev/LongTrain.ipynb
dev/Mask.ipynb
dev/Memory copy.ipynb
dev/Memory.ipynb
dev/OptimalSequence.ipynb
dev/PlotsQualificacao.ipynb
dev/ProtocolTest.ipynb
dev/RandomGenerator.ipynb
dev/Refact.ipynb
dev/STR.ipynb
dev/SensorialTrain.ipynb
dev/StateDiscover.ipynb
dev/Toy1D_Exp2.ipynb
dev/Toy1d.ipynb
dev/Toy1dData2.ipynb
dev/Toy1d_Metrics2.ipynb
dev/check_conversion.ipynb
dev/compile.ipynb
dev/cos.ipynb
dev/cosin_test.py
dev/dataset_figures.ipynb
dev/export_pipeline.ipynb
dev/fix_sdtw.ipynb
dev/hamilton_parameterize.ipynb
dev/local_test.py
dev/profile_optimize.ipynb
dev/target_mask.ipynb
dev/test_metrics.py
dev/toy1d_autoregressive.py
dev/toy1d_autoregressive_mask_sensorial.py
dev/toy1d_block_configuration.py
dev/toy1d_discover_state.py
dev/toy1d_export.ipynb
dev/toy1d_export.py
dev/toy1d_long_training.py
dev/toy1d_mask_sensorial.py
dev/toy1d_memory.py
dev/toy1d_post_train.ipynb
dev/toy1d_simple_shift.py
dev/tqdm_test.py
dev/zip_test.ipynb
dev/structural_break/Data.ipynb
dev/structural_break/model_test.ipynb
dev/toy1d_experiments/configuration_test_linear_model.ipynb
dev/toy1d_experiments/configutation_test.ipynb
dev/toy1d_experiments/configutation_test_stats.ipynb
dev/toy1d_experiments/protocol_test.ipynb
examples/My First World Machine.ipynb
experiments/.gitignore
experiments/LICENSE
experiments/README.md
experiments/pyproject.toml
experiments/rename.ipynb
experiments/reports/World Machine - Toy1D - Experiment 0 Protocol Test.md
experiments/reports/World Machine - Toy1D - Experiment 1 Configuration Test.md
experiments/reports/World Machine - Toy1D - Experiment 2 Best Long.md
experiments/reports/World Machine - Toy1D Dataset.md
experiments/reports/toy1d_experiment0/metrics_box_plot_0.1sdtw.png
experiments/reports/toy1d_experiment0/metrics_box_plot_mse.png
experiments/reports/toy1d_experiment0/pipeline_base.svg
experiments/reports/toy1d_experiment0/pipeline_experiment.svg
experiments/reports/toy1d_experiment0/pipeline_multiple.svg
experiments/reports/toy1d_experiment0/pipeline_variation.svg
experiments/reports/toy1d_experiment0/state_decoded_samples.png
experiments/reports/toy1d_experiment0/toy1d_samples.png
experiments/reports/toy1d_experiment0/train_history_optimizer_loss.png
experiments/reports/toy1d_experiment1/best_models_metrics.md
experiments/reports/toy1d_experiment1/divergence_probability.png
experiments/reports/toy1d_experiment1/filtered_divergence_probability.png
experiments/reports/toy1d_experiment1/joint_impact_normal.png
experiments/reports/toy1d_experiment1/joint_impact_prediction.png
experiments/reports/toy1d_experiment1/joint_impact_prediction_local.png
experiments/reports/toy1d_experiment1/joint_impact_prediction_shallow.png
experiments/reports/toy1d_experiment1/joint_impact_use_state.png
experiments/reports/toy1d_experiment1/metric_distribution_mse.png
experiments/reports/toy1d_experiment1/pearson_mse_between_tasks.png
experiments/reports/toy1d_experiment1/pipeline_base.svg
experiments/reports/toy1d_experiment1/pipeline_final.svg
experiments/reports/toy1d_experiment1/pipeline_final_rotated.svg
experiments/reports/toy1d_experiment1/pipeline_multiple.svg
experiments/reports/toy1d_experiment1/pipeline_variation.svg
experiments/reports/toy1d_experiment1/variable_impact_on_duration.png
experiments/reports/toy1d_experiment1/variable_impact_on_tasks.png
experiments/reports/toy1d_experiment2/optimizer_loss.png
experiments/reports/toy1d_experiment2/pipeline_base.svg
experiments/reports/toy1d_experiment2/pipeline_final.svg
experiments/reports/toy1d_experiment2/pipeline_multiple.svg
experiments/reports/toy1d_experiment2/pipeline_variation.svg
experiments/reports/toy1d_experiment2/prediction_shallow_samples.png
experiments/reports/toy1d_experiment2/variation_performance.png
experiments/src/world_machine_experiments/__init__.py
experiments/src/world_machine_experiments/shared/__init__.py
experiments/src/world_machine_experiments/shared/acronyms.py
experiments/src/world_machine_experiments/shared/function_variation.py
experiments/src/world_machine_experiments/shared/load_multiple_metrics.py
experiments/src/world_machine_experiments/shared/metric_plot.py
experiments/src/world_machine_experiments/shared/parameter_variation_plots.py
experiments/src/world_machine_experiments/shared/pipeline.py
experiments/src/world_machine_experiments/shared/random.py
experiments/src/world_machine_experiments/shared/save_metrics.py
experiments/src/world_machine_experiments/shared/save_parameters.py
experiments/src/world_machine_experiments/shared/save_plots.py
experiments/src/world_machine_experiments/shared/split_data.py
experiments/src/world_machine_experiments/shared/statistics.py
experiments/src/world_machine_experiments/shared/train_plots.py
experiments/src/world_machine_experiments/tools/__init__.py
experiments/src/world_machine_experiments/tools/export_zip.py
experiments/src/world_machine_experiments/tools/zenodo_upload.py
experiments/src/world_machine_experiments/toy1d/__init__.py
experiments/src/world_machine_experiments/toy1d/channels.py
experiments/src/world_machine_experiments/toy1d/experiment0_protocol_test.py
experiments/src/world_machine_experiments/toy1d/experiment1_configuration_test.py
experiments/src/world_machine_experiments/toy1d/experiment2_best_long.py
experiments/src/world_machine_experiments/toy1d/export_dataset.py
experiments/src/world_machine_experiments/toy1d/base/__init__.py
experiments/src/world_machine_experiments/toy1d/base/autoregressive.py
experiments/src/world_machine_experiments/toy1d/base/data.py
experiments/src/world_machine_experiments/toy1d/base/dataloader.py
experiments/src/world_machine_experiments/toy1d/base/dataset.py
experiments/src/world_machine_experiments/toy1d/base/mask_sensory.py
experiments/src/world_machine_experiments/toy1d/base/metrics.py
experiments/src/world_machine_experiments/toy1d/base/model.py
experiments/src/world_machine_experiments/toy1d/base/plot.py
experiments/src/world_machine_experiments/toy1d/base/simple_shift_loss.py
experiments/src/world_machine_experiments/toy1d/base/train.py
experiments/src/world_machine_experiments/toy1d/multiple/__init__.py
experiments/src/world_machine_experiments/toy1d/multiple/multiple_runs.py
experiments/src/world_machine_experiments/toy1d/parameter_variation/__init__.py
experiments/src/world_machine_experiments/toy1d/parameter_variation/parameter_variation.py
experiments/src/world_machine_experiments/toy1d/specific/__init__.py
experiments/src/world_machine_experiments/toy1d/specific/experiment0.py
experiments/src/world_machine_experiments/toy1d/specific/experiment2.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/__init__.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/_shared.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/best_models.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/disjoint_groups.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/divergence_probability.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/impact_test.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/joint_impact_test.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/masked_percentage.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/metrics_distribution.py
experiments/src/world_machine_experiments/toy1d/specific/experiment1/prepare_data.py
page/Makefile
page/make.bat
page/pyproject.toml
page/source/conf.py
page/source/index.rst
page/source/_static/artifact_explorer.js
page/source/_static/artifacts.json
page/source/_static/custom-icons.js
page/source/_static/d3.v7.min.js
page/source/artifact_explorer/index.md
page/source/blog/index.md
page/source/blog/post_test.md
page/source/examples/index.md
page/source/reports/index.md
page/src/world_machine_page/__init__.py
page/src/world_machine_page/html_theme_options.py
page/src/world_machine_page/post_build.py
page/src/world_machine_page/pre_build.py
page/src/world_machine_page/utils.py
src/world_machine/__init__.py
src/world_machine/current_version.py
src/world_machine/wm_builder.py
src/world_machine/world_machine.py
src/world_machine.egg-info/PKG-INFO
src/world_machine.egg-info/SOURCES.txt
src/world_machine.egg-info/dependency_links.txt
src/world_machine.egg-info/requires.txt
src/world_machine.egg-info/top_level.txt
src/world_machine/data/__init__.py
src/world_machine/data/dataloader.py
src/world_machine/data/dataset.py
src/world_machine/evaluate/__init__.py
src/world_machine/evaluate/metrics_generator.py
src/world_machine/layers/__init__.py
src/world_machine/layers/adaln_zero_block.py
src/world_machine/layers/attention.py
src/world_machine/layers/block_container.py
src/world_machine/layers/clamp.py
src/world_machine/layers/conditioning_block.py
src/world_machine/layers/ltanh.py
src/world_machine/layers/modulate.py
src/world_machine/layers/pointwise_feedforward.py
src/world_machine/layers/sin_tanh.py
src/world_machine/layers/sine.py
src/world_machine/layers/transform_decoder_block.py
src/world_machine/layers/positional_encoder/__init__.py
src/world_machine/layers/positional_encoder/alibi.py
src/world_machine/layers/positional_encoder/create.py
src/world_machine/layers/positional_encoder/positional_encoder.py
src/world_machine/layers/positional_encoder/sine.py
src/world_machine/profile/__init__.py
src/world_machine/profile/profile_range.py
src/world_machine/profile/nvtx/__init__.py
src/world_machine/profile/nvtx/annotate_se.py
src/world_machine/profile/nvtx/colors.py
src/world_machine/profile/nvtx/mock.py
src/world_machine/train/__init__.py
src/world_machine/train/criterion_set.py
src/world_machine/train/mode.py
src/world_machine/train/scheduler.py
src/world_machine/train/trainer.py
src/world_machine/train/stages/__init__.py
src/world_machine/train/stages/early_stopper.py
src/world_machine/train/stages/gradient_accumulator.py
src/world_machine/train/stages/lambda_stage.py
src/world_machine/train/stages/local_setter.py
src/world_machine/train/stages/loss_manager.py
src/world_machine/train/stages/lrscheduler_step.py
src/world_machine/train/stages/noise_adder.py
src/world_machine/train/stages/prepare_model.py
src/world_machine/train/stages/sensory_masker.py
src/world_machine/train/stages/sequence_breaker.py
src/world_machine/train/stages/short_time_recaller.py
src/world_machine/train/stages/simple_optimizer.py
src/world_machine/train/stages/state_manager.py
src/world_machine/train/stages/train_stage.py
src/world_machine/train/stages/world_machine_forward.py
src/world_machine/version_upgrader/__init__.py
src/world_machine/version_upgrader/upgrade_none_to_1.py
src/world_machine/version_upgrader/upgrader.py