LICENSE
MANIFEST.in
README.md
pyproject.toml
config/__init__.py
config/generate_latent_dataset.yaml
config/main.yaml
config/dataset/celeba.yaml
config/dataset/cifar10.yaml
config/dataset/counting_polygons_blank.yaml
config/dataset/counting_polygons_blank_ambiguous_conditional.yaml
config/dataset/counting_polygons_blank_explicit_conditional.yaml
config/dataset/counting_polygons_ffhq.yaml
config/dataset/counting_polygons_ffhq_ambiguous_conditional.yaml
config/dataset/counting_polygons_ffhq_explicit_conditional.yaml
config/dataset/counting_stars_blank.yaml
config/dataset/counting_stars_blank_no_nums.yaml
config/dataset/counting_stars_ffhq.yaml
config/dataset/counting_stars_ffhq_no_nums.yaml
config/dataset/even_pixels.yaml
config/dataset/ffhq.yaml
config/dataset/imagenet256.yaml
config/dataset/imagenet64.yaml
config/dataset/mnist.yaml
config/dataset/mnist_sudoku_top1000.yaml
config/dataset/mnist_sudoku_top1000_easy.yaml
config/dataset/mnist_sudoku_top1000_hard.yaml
config/dataset/mnist_sudoku_top1000_medium.yaml
config/dataset/realestate10k.yaml
config/denoising_model/denoiser/denoiser.yaml
config/denoising_model/denoiser/dit.yaml
config/denoising_model/denoiser/dit_b_2.yaml
config/denoising_model/denoiser/dit_l_2.yaml
config/denoising_model/denoiser/dit_s_2.yaml
config/denoising_model/denoiser/dit_xl_2.yaml
config/denoising_model/denoiser/lightningdit.yaml
config/denoising_model/denoiser/lightningdit_xl.yaml
config/denoising_model/denoiser/mar.yaml
config/denoising_model/denoiser/mar_b.yaml
config/denoising_model/denoiser/mar_h.yaml
config/denoising_model/denoiser/mar_l.yaml
config/denoising_model/denoiser/spiral_mlp.yaml
config/denoising_model/denoiser/u_vit3d_pose.yaml
config/denoising_model/denoiser/u_vit3d_pose_sm.yaml
config/denoising_model/denoiser/unet.yaml
config/denoising_model/denoiser/unet_128.yaml
config/denoising_model/denoiser/unet_128_imagenet.yaml
config/denoising_model/denoiser/unet_128_lite.yaml
config/denoising_model/denoiser/unet_256.yaml
config/denoising_model/denoiser/unet_28.yaml
config/denoising_model/denoiser/unet_32.yaml
config/denoising_model/denoiser/unet_32_cifar10.yaml
config/denoising_model/denoiser/unet_32_ldm.yaml
config/denoising_model/denoiser/unet_32_lite.yaml
config/denoising_model/denoiser/unet_64.yaml
config/denoising_model/denoiser/unet_64_imagenet.yaml
config/denoising_model/denoiser/xar.yaml
config/denoising_model/denoiser/xar_b.yaml
config/denoising_model/denoiser/xar_h.yaml
config/denoising_model/denoiser/xar_l.yaml
config/denoising_model/denoiser/class_embedding/parameters.yaml
config/denoising_model/denoiser/embedding/sinusodial.yaml
config/denoising_model/flow/continuous_cosine_logsnr.yaml
config/denoising_model/flow/cosine.yaml
config/denoising_model/flow/diffusion.yaml
config/denoising_model/flow/original_diffusion.yaml
config/denoising_model/flow/rectified.yaml
config/denoising_model/flow/beta_schedule/cosine.yaml
config/denoising_model/flow/beta_schedule/cosine_simple_diffusion.yaml
config/denoising_model/flow/beta_schedule/linear.yaml
config/experiment/api_default.yaml
config/experiment/even_pixels.yaml
config/experiment/mnist_sudoku.yaml
config/experiment/mnist_sudoku_unet.yaml
config/experiment/mnist_sudoku_unet_base.yaml
config/experiment/mnist_sudoku_unet_mb_0.7.yaml
config/experiment/mnist_sudoku_unet_mb_1.yaml
config/experiment/mnist_sudoku_unet_trainable_weights.yaml
config/experiment/realestate10k.yaml
config/experiment/benchmark/counting_polygons_blank_1000.yaml
config/experiment/benchmark/even_pixels_100.yaml
config/experiment/benchmark/even_pixels_1000.yaml
config/experiment/benchmark/mnist_sudoku_top1000.yaml
config/experiment/benchmark/mnist_sudoku_top1000_easy.yaml
config/experiment/benchmark/mnist_sudoku_top1000_hard.yaml
config/experiment/benchmark/mnist_sudoku_top1000_medium.yaml
config/experiment/benchmark/none.yaml
config/experiment/benchmark/realestate10k_20.yaml
config/inference_sampler/inference_sampler.yaml
config/inference_sampler/scheduled_sampler.yaml
config/inference_sampler/cfg_scheduler/constant.yaml
config/inference_sampler/cfg_scheduler/linear.yaml
config/inference_sampler/sampling_schedule/adaptive_inverse.yaml
config/inference_sampler/sampling_schedule/adaptive_sequential.yaml
config/inference_sampler/sampling_schedule/adaptive_softmax.yaml
config/inference_sampler/sampling_schedule/fixed.yaml
config/inference_sampler/sampling_schedule/graph_sequential.yaml
config/inference_sampler/sampling_schedule/mask_git.yaml
config/inference_sampler/sampling_schedule/random_sequential.yaml
config/inference_sampler/sampling_schedule/raster.yaml
config/optimizer/default.yaml
config/optimizer/dit.yaml
config/optimizer/mar_tune.yaml
config/optimizer/vrm.yaml
config/time_sampler/independent.yaml
config/time_sampler/mean_beta.yaml
config/time_sampler/mean_beta_trainable.yaml
config/time_sampler/sequential.yaml
config/time_sampler/shared.yaml
config/time_sampler/trainable.yaml
config/time_sampler/scalar_time_sampler/logit_normal.yaml
config/time_sampler/scalar_time_sampler/uniform.yaml
config/time_sampler/scalar_time_sampler/uniform_trimmed.yaml
config/time_sampler/time_weighting/logit_normal.yaml
config/time_sampler/time_weighting/uniform.yaml
config/variable_mapper/image.yaml
config/variable_mapper/video_pose.yaml
config/variable_mapper/autoencoder/kl_16.yaml
config/variable_mapper/autoencoder/kl_16_mar.yaml
config/variable_mapper/autoencoder/kl_16_vavae.yaml
config/variable_mapper/autoencoder/kl_8.yaml
spatialreasoners/__init__.py
spatialreasoners/_main.py
spatialreasoners/api.py
spatialreasoners/config.py
spatialreasoners/env.py
spatialreasoners/global_cfg.py
spatialreasoners/main.py
spatialreasoners/registry.py
spatialreasoners/type_extensions.py
spatialreasoners.egg-info/PKG-INFO
spatialreasoners.egg-info/SOURCES.txt
spatialreasoners.egg-info/dependency_links.txt
spatialreasoners.egg-info/requires.txt
spatialreasoners.egg-info/top_level.txt
spatialreasoners/benchmark/__init__.py
spatialreasoners/benchmark/benchmark.py
spatialreasoners/benchmark/evaluation/__init__.py
spatialreasoners/benchmark/evaluation/evaluation.py
spatialreasoners/benchmark/evaluation/sampling_evaluation.py
spatialreasoners/benchmark/evaluation/type_extensions.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/__init__.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/frame_vis_sampling_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/video_pose_sampling_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/__init__.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/counting_polygons_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/even_pixels_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/image_sampling_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/mnist_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/mnist_grid_evaluation.py
spatialreasoners/benchmark/evaluation/frame_vis_sampling_evaluation/image_sampling_evaluation/mnist_sudoku_evaluation.py
spatialreasoners/benchmark/evaluation/inference_sampler/__init__.py
spatialreasoners/benchmark/evaluation/inference_sampler/inference_sampler.py
spatialreasoners/benchmark/evaluation/inference_sampler/scheduled_inference_sampler.py
spatialreasoners/benchmark/evaluation/inference_sampler/type_extensions.py
spatialreasoners/benchmark/evaluation/inference_sampler/cfg_scheduler/__init__.py
spatialreasoners/benchmark/evaluation/inference_sampler/cfg_scheduler/cfg_scheduler.py
spatialreasoners/benchmark/evaluation/inference_sampler/cfg_scheduler/constant.py
spatialreasoners/benchmark/evaluation/inference_sampler/cfg_scheduler/linear.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/__init__.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/fixed.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/mask_git.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/raster.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/sampling_schedule.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/adaptive/__init__.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/adaptive/adaptive.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/adaptive/adaptive_inverse.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/adaptive/adaptive_softmax.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/certainty_sequential/__init__.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/certainty_sequential/adaptive_sequential.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/certainty_sequential/certainty_sequential.py
spatialreasoners/benchmark/evaluation/inference_sampler/sampling_schedule/certainty_sequential/graph_sequential.py
spatialreasoners/dataset/__init__.py
spatialreasoners/dataset/dataset.py
spatialreasoners/dataset/dataset_image/__init__.py
spatialreasoners/dataset/dataset_image/dataset_celeba.py
spatialreasoners/dataset/dataset_image/dataset_cifar10.py
spatialreasoners/dataset/dataset_image/dataset_even_pixels.py
spatialreasoners/dataset/dataset_image/dataset_ffhq.py
spatialreasoners/dataset/dataset_image/dataset_grid.py
spatialreasoners/dataset/dataset_image/dataset_image.py
spatialreasoners/dataset/dataset_image/dataset_imagenet.py
spatialreasoners/dataset/dataset_image/dataset_mnist.py
spatialreasoners/dataset/dataset_image/dataset_mnist_grid_lazy.py
spatialreasoners/dataset/dataset_image/dataset_mnist_sudoku_9x9_lazy.py
spatialreasoners/dataset/dataset_image/latent_folder.py
spatialreasoners/dataset/dataset_image/type_extensions.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/__init__.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/dataset_counting_polygons_base.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/dataset_counting_polygons_blank.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/dataset_counting_polygons_ffhq.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/dataset_counting_polygons_subdataset.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/font_cache.py
spatialreasoners/dataset/dataset_image/dataset_counting_polygons/labelers.py
spatialreasoners/dataset/dataset_video/__init__.py
spatialreasoners/dataset/dataset_video/dataset_realestate10k.py
spatialreasoners/dataset/dataset_video/dataset_video.py
spatialreasoners/dataset/dataset_video/geometry_utils.py
spatialreasoners/dataset/dataset_video/type_extensions.py
spatialreasoners/dataset/dataset_video/video_transform.py
spatialreasoners/denoising_model/__init__.py
spatialreasoners/denoising_model/denoising_model.py
spatialreasoners/denoising_model/type_extensions.py
spatialreasoners/denoising_model/denoiser/__init__.py
spatialreasoners/denoising_model/denoiser/activations.py
spatialreasoners/denoising_model/denoiser/denoiser.py
spatialreasoners/denoising_model/denoiser/norm_layers.py
spatialreasoners/denoising_model/denoiser/utils.py
spatialreasoners/denoising_model/denoiser/class_embedding/__init__.py
spatialreasoners/denoising_model/denoiser/class_embedding/class_embedding.py
spatialreasoners/denoising_model/denoiser/class_embedding/parameters.py
spatialreasoners/denoising_model/denoiser/dit/__init__.py
spatialreasoners/denoising_model/denoiser/dit/model.py
spatialreasoners/denoising_model/denoiser/dit/type_extensions.py
spatialreasoners/denoising_model/denoiser/dit/modules/__init__.py
spatialreasoners/denoising_model/denoiser/dit/modules/dit_block.py
spatialreasoners/denoising_model/denoiser/dit/modules/output_layer.py
spatialreasoners/denoising_model/denoiser/dit/modules/patch_embedding.py
spatialreasoners/denoising_model/denoiser/embedding/__init__.py
spatialreasoners/denoising_model/denoiser/embedding/embedding.py
spatialreasoners/denoising_model/denoiser/embedding/sinusodial.py
spatialreasoners/denoising_model/denoiser/modules/attention.py
spatialreasoners/denoising_model/denoiser/modules/vit_block.py
spatialreasoners/denoising_model/denoiser/modules/feed_forward/__init__.py
spatialreasoners/denoising_model/denoiser/modules/feed_forward/feed_forward.py
spatialreasoners/denoising_model/denoiser/modules/feed_forward/lightningdit_swiglu.py
spatialreasoners/denoising_model/denoiser/modules/feed_forward/mlp.py
spatialreasoners/denoising_model/denoiser/modules/feed_forward/timm_swiglu.py
spatialreasoners/denoising_model/denoiser/modules/norm_1d/__init__.py
spatialreasoners/denoising_model/denoiser/modules/norm_1d/norm_1d.py
spatialreasoners/denoising_model/denoiser/modules/norm_2d/__init__.py
spatialreasoners/denoising_model/denoiser/modules/norm_2d/norm_2d.py
spatialreasoners/denoising_model/denoiser/modules/pos_embedding/__init__.py
spatialreasoners/denoising_model/denoiser/modules/pos_embedding/frequency.py
spatialreasoners/denoising_model/denoiser/modules/pos_embedding/pos_embedding.py
spatialreasoners/denoising_model/denoiser/modules/pos_embedding/rotary.py
spatialreasoners/denoising_model/denoiser/u_vit3d/__init__.py
spatialreasoners/denoising_model/denoiser/u_vit3d/base_backbone.py
spatialreasoners/denoising_model/denoiser/u_vit3d/type_extensions.py
spatialreasoners/denoising_model/denoiser/u_vit3d/u_vit3d.py
spatialreasoners/denoising_model/denoiser/u_vit3d/u_vit3d_pose.py
spatialreasoners/denoising_model/denoiser/u_vit3d/u_vit_blocks.py
spatialreasoners/denoising_model/denoiser/u_vit3d/modules/__init__.py
spatialreasoners/denoising_model/denoiser/u_vit3d/modules/embeddings.py
spatialreasoners/denoising_model/denoiser/u_vit3d/modules/normalization.py
spatialreasoners/denoising_model/denoiser/u_vit3d/modules/zero_module.py
spatialreasoners/denoising_model/denoiser/unet/__init__.py
spatialreasoners/denoising_model/denoiser/unet/model.py
spatialreasoners/denoising_model/denoiser/unet/type_extensions.py
spatialreasoners/denoising_model/denoiser/unet/modules/__init__.py
spatialreasoners/denoising_model/denoiser/unet/modules/bottleneck.py
spatialreasoners/denoising_model/denoiser/unet/modules/decoder.py
spatialreasoners/denoising_model/denoiser/unet/modules/downsample.py
spatialreasoners/denoising_model/denoiser/unet/modules/embed_sequential.py
spatialreasoners/denoising_model/denoiser/unet/modules/encoder.py
spatialreasoners/denoising_model/denoiser/unet/modules/multi_head_attention.py
spatialreasoners/denoising_model/denoiser/unet/modules/res_block.py
spatialreasoners/denoising_model/denoiser/unet/modules/upsample.py
spatialreasoners/denoising_model/denoiser/xar/__init__.py
spatialreasoners/denoising_model/denoiser/xar/model.py
spatialreasoners/denoising_model/denoiser/xar/type_extensions.py
spatialreasoners/denoising_model/denoiser/xar/modules/__init__.py
spatialreasoners/denoising_model/denoiser/xar/modules/xar_block.py
spatialreasoners/denoising_model/flow/__init__.py
spatialreasoners/denoising_model/flow/continuous_diffusion_flow.py
spatialreasoners/denoising_model/flow/cosine_flow.py
spatialreasoners/denoising_model/flow/diagonal_gaussian.py
spatialreasoners/denoising_model/flow/diffusion.py
spatialreasoners/denoising_model/flow/flow.py
spatialreasoners/denoising_model/flow/original_diffusion.py
spatialreasoners/denoising_model/flow/rectified_flow.py
spatialreasoners/denoising_model/flow/beta_schedule/__init__.py
spatialreasoners/denoising_model/flow/beta_schedule/beta_schedule.py
spatialreasoners/denoising_model/flow/beta_schedule/cosine.py
spatialreasoners/denoising_model/flow/beta_schedule/cosine_simple_diffusion.py
spatialreasoners/denoising_model/flow/beta_schedule/linear.py
spatialreasoners/denoising_model/tokenizer/__init__.py
spatialreasoners/denoising_model/tokenizer/tokenizer.py
spatialreasoners/denoising_model/tokenizer/dit_tokenizer/__init__.py
spatialreasoners/denoising_model/tokenizer/dit_tokenizer/dit_tokenizer.py
spatialreasoners/denoising_model/tokenizer/dit_tokenizer/image_dit_tokenizer.py
spatialreasoners/denoising_model/tokenizer/u_vit_tokenizer/__init__.py
spatialreasoners/denoising_model/tokenizer/u_vit_tokenizer/pose_video_u_vit_tokenizer.py
spatialreasoners/denoising_model/tokenizer/unet_tokenizer/__init__.py
spatialreasoners/denoising_model/tokenizer/unet_tokenizer/unet_tokenizer.py
spatialreasoners/misc/counting_polygons_classifier.py
spatialreasoners/misc/dict_utils.py
spatialreasoners/misc/image_io.py
spatialreasoners/misc/latent_saver.py
spatialreasoners/misc/local_logger.py
spatialreasoners/misc/mnist_classifier.py
spatialreasoners/misc/nn_module_tools.py
spatialreasoners/misc/step_tracker.py
spatialreasoners/misc/tensor.py
spatialreasoners/misc/time_tester.py
spatialreasoners/misc/utils.py
spatialreasoners/misc/wandb_tools.py
spatialreasoners/scripts/dit_ckpt_mapping.py
spatialreasoners/scripts/lightningdit_ckpt_mapping.py
spatialreasoners/scripts/mar_ckpt_mapping.py
spatialreasoners/scripts/vae_ckpt_mapping.py
spatialreasoners/scripts/xar_ckpt_mapping.py
spatialreasoners/training/__init__.py
spatialreasoners/training/data_module.py
spatialreasoners/training/srm_lightning_module.py
spatialreasoners/training/loss/__init__.py
spatialreasoners/training/loss/cosine.py
spatialreasoners/training/loss/loss.py
spatialreasoners/training/loss/mse.py
spatialreasoners/training/loss/nll.py
spatialreasoners/training/loss/vlb.py
spatialreasoners/training/time_sampler/__init__.py
spatialreasoners/training/time_sampler/independent.py
spatialreasoners/training/time_sampler/mean_beta.py
spatialreasoners/training/time_sampler/mean_simple.py
spatialreasoners/training/time_sampler/sequential.py
spatialreasoners/training/time_sampler/shared.py
spatialreasoners/training/time_sampler/time_sampler.py
spatialreasoners/training/time_sampler/two_stage_time_sampler.py
spatialreasoners/training/time_sampler/mask_ratio_sampler/__init__.py
spatialreasoners/training/time_sampler/mask_ratio_sampler/mask_ratio_sampler.py
spatialreasoners/training/time_sampler/mask_ratio_sampler/trunc_norm.py
spatialreasoners/training/time_sampler/mask_ratio_sampler/uniform.py
spatialreasoners/training/time_sampler/normalization_weighing/__init__.py
spatialreasoners/training/time_sampler/normalization_weighing/histogram_pdf_weighing.py
spatialreasoners/training/time_sampler/normalization_weighing/normalization_weighing.py
spatialreasoners/training/time_sampler/normalization_weighing/trainable.py
spatialreasoners/training/time_sampler/scalar_time_sampler/__init__.py
spatialreasoners/training/time_sampler/scalar_time_sampler/logit_normal.py
spatialreasoners/training/time_sampler/scalar_time_sampler/mode.py
spatialreasoners/training/time_sampler/scalar_time_sampler/scalar_time_sampler.py
spatialreasoners/training/time_sampler/scalar_time_sampler/uniform.py
spatialreasoners/training/time_sampler/time_weighting/__init__.py
spatialreasoners/training/time_sampler/time_weighting/cosmap.py
spatialreasoners/training/time_sampler/time_weighting/logit_normal.py
spatialreasoners/training/time_sampler/time_weighting/sigma_sqrt.py
spatialreasoners/training/time_sampler/time_weighting/time_weighting.py
spatialreasoners/variable_mapper/__init__.py
spatialreasoners/variable_mapper/image_variable_mapper.py
spatialreasoners/variable_mapper/sudoku_variable_mapper.py
spatialreasoners/variable_mapper/variable_mapper.py
spatialreasoners/variable_mapper/video_pose_variable_mapper.py
spatialreasoners/variable_mapper/autoencoder/__init__.py
spatialreasoners/variable_mapper/autoencoder/autoencoder.py
spatialreasoners/variable_mapper/autoencoder/image_autoencoder.py
spatialreasoners/variable_mapper/autoencoder/kl_image_autoencoder.py
spatialreasoners/visualization/annotation.py
spatialreasoners/visualization/color_map.py
spatialreasoners/visualization/layout.py