.dockerignore
.gitattributes
.gitignore
delete_bin.ipynb
dockerfile
pyproject.toml
.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
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/compile.ipynb
dev/cos.ipynb
dev/cosin_test.py
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_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/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
experiments/.gitignore
experiments/README.md
experiments/pyproject.toml
experiments/rename.ipynb
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/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/toy1d/__init__.py
experiments/src/world_machine_experiments/toy1d/dimensions.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/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_sensorial.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/experiment0.py
src/world_machine/__init__.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/noise_adder.py
src/world_machine/train/stages/prepare_model.py
src/world_machine/train/stages/sensorial_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