.gitattributes
.gitignore
.pre-commit-config.yaml
LICENSE
MANIFEST.in
NOTICE
README.md
WEIGHTS_LICENSE
hubconf.py
pyproject.toml
requirements.txt
setup.cfg
setup.py
.github/CODE_OF_CONDUCT.md
.github/CONTRIBUTING.md
.github/PULL_REQUEST_TEMPLATE.md
.github/ISSUE_TEMPLATE/bug_template.md
.github/ISSUE_TEMPLATE/config.yml
.github/ISSUE_TEMPLATE/enhancement_template.md
.github/workflows/build-and-test.yml
.github/workflows/publish-package.yml
neuralcompression/__init__.py
neuralcompression/version.py
neuralcompression.egg-info/PKG-INFO
neuralcompression.egg-info/SOURCES.txt
neuralcompression.egg-info/dependency_links.txt
neuralcompression.egg-info/requires.txt
neuralcompression.egg-info/top_level.txt
neuralcompression/_outputs/__init__.py
neuralcompression/_outputs/_discriminator_output.py
neuralcompression/_outputs/_hyperprior_output.py
neuralcompression/_outputs/_vqvae_autoencoder_output.py
neuralcompression/data/__init__.py
neuralcompression/data/_clic_2020_image.py
neuralcompression/data/_clic_2020_video.py
neuralcompression/data/_div2k_dataset.py
neuralcompression/data/_kodak.py
neuralcompression/data/_openimages_v6.py
neuralcompression/data/_vimeo_90k_septuplet.py
neuralcompression/distributions/__init__.py
neuralcompression/distributions/_noisy_normal.py
neuralcompression/distributions/_uniform_noise.py
neuralcompression/functional/__init__.py
neuralcompression/functional/_count_flops.py
neuralcompression/functional/_dense_image_warp.py
neuralcompression/functional/_estimate_tails.py
neuralcompression/functional/_hsv_to_rgb.py
neuralcompression/functional/_image_to_255_scale.py
neuralcompression/functional/_information_content.py
neuralcompression/functional/_log_cdf.py
neuralcompression/functional/_log_expm1.py
neuralcompression/functional/_log_ndtr.py
neuralcompression/functional/_log_survival_function.py
neuralcompression/functional/_lower_bound.py
neuralcompression/functional/_lower_tail.py
neuralcompression/functional/_lpips.py
neuralcompression/functional/_multiscale_structural_similarity.py
neuralcompression/functional/_ndtr.py
neuralcompression/functional/_optical_flow_to_color.py
neuralcompression/functional/_pad_image.py
neuralcompression/functional/_quantization_offset.py
neuralcompression/functional/_soft_round.py
neuralcompression/functional/_soft_round_conditional_mean.py
neuralcompression/functional/_soft_round_inverse.py
neuralcompression/functional/_survival_function.py
neuralcompression/functional/_upper_tail.py
neuralcompression/layers/__init__.py
neuralcompression/layers/_channel_norm.py
neuralcompression/layers/_continuous_entropy.py
neuralcompression/layers/_normalize_latent.py
neuralcompression/layers/_project_latent.py
neuralcompression/layers/_q_bottleneck.py
neuralcompression/layers/_simplified_gdn.py
neuralcompression/layers/_vq_bottleneck.py
neuralcompression/layers/_xca.py
neuralcompression/loss_fn/__init__.py
neuralcompression/loss_fn/_binary_crossentropy_loss.py
neuralcompression/loss_fn/_distortion_loss.py
neuralcompression/loss_fn/_gan_losses.py
neuralcompression/loss_fn/_mse_loss.py
neuralcompression/loss_fn/_mse_lpips_loss.py
neuralcompression/loss_fn/_normfix_lpips.py
neuralcompression/loss_fn/_oasis_loss.py
neuralcompression/loss_fn/_target_rate_config.py
neuralcompression/metrics/__init__.py
neuralcompression/metrics/_calc_psnr.py
neuralcompression/metrics/_dists.py
neuralcompression/metrics/_fid.py
neuralcompression/metrics/_fid_swav.py
neuralcompression/metrics/_kid.py
neuralcompression/metrics/_msssim.py
neuralcompression/metrics/_pickle_size.py
neuralcompression/metrics/_update_patch_fid.py
neuralcompression/models/__init__.py
neuralcompression/models/_discriminator.py
neuralcompression/models/_hific_autoencoder.py
neuralcompression/models/_hific_encoder_decoder.py
neuralcompression/models/_hyperprior_autoencoder.py
neuralcompression/models/_unet_discriminator.py
neuralcompression/models/_vqvae_projector.py
neuralcompression/models/_vqvae_xcit_autoencoder.py
neuralcompression/models/deep_video_compression.py
neuralcompression/optim/__init__.py
neuralcompression/optim/_identity.py
neuralcompression/optim/_ramp_const_cosine.py
neuralcompression/optim/_ramp_cosine.py
neuralcompression/zoo/__init__.py
neuralcompression/zoo/_msillm.py
neuralcompression/zoo/_msillm_vqvae.py
neuralcompression/zoo/_noganms.py
projects/README.md
projects/bits_back_diffusion/.gitignore
projects/bits_back_diffusion/README.md
projects/bits_back_diffusion/compute_vlb.py
projects/bits_back_diffusion/encode.py
projects/bits_back_diffusion/environment.yml
projects/bits_back_diffusion/test.py
projects/bits_back_diffusion/train.py
projects/bits_back_diffusion/bits_back_diffusion/__init__.py
projects/bits_back_diffusion/bits_back_diffusion/codec.py
projects/bits_back_diffusion/bits_back_diffusion/script_util.py
projects/bits_back_diffusion/config/compute_vlb.yaml
projects/bits_back_diffusion/config/default.yaml
projects/bits_back_diffusion/config/encode.yaml
projects/bits_back_diffusion/config/train.yaml
projects/bits_back_diffusion/config/data/cifar.yaml
projects/bits_back_diffusion/config/data/imagenet.yaml
projects/bits_back_diffusion/config/diffusion/default.yaml
projects/bits_back_diffusion/config/diffusion/respaced.yaml
projects/bits_back_diffusion/config/evaluator/cifar.yaml
projects/bits_back_diffusion/config/evaluator/imagenet.yaml
projects/bits_back_diffusion/config/mode/submitit.yaml
projects/bits_back_diffusion/config/model/cifar.yaml
projects/bits_back_diffusion/config/model/imagenet.yaml
projects/bits_back_diffusion/datasets/cifar10.py
projects/deep_video_compression/README.md
projects/deep_video_compression/_utils.py
projects/deep_video_compression/data_module.py
projects/deep_video_compression/dvc_module.py
projects/deep_video_compression/requirements.txt
projects/deep_video_compression/train.py
projects/deep_video_compression/config/base.yaml
projects/deep_video_compression/config/mode/submitit_multi_node.yaml
projects/deep_video_compression/config/mode/submitit_single_node.yaml
projects/deep_video_compression/config/training_stages/s1_motion_estimation.yaml
projects/deep_video_compression/config/training_stages/s2_motion_compression.yaml
projects/deep_video_compression/config/training_stages/s3_motion_compensation.yaml
projects/deep_video_compression/config/training_stages/s4_total_2frame.yaml
projects/deep_video_compression/config/training_stages/s5_total.yaml
projects/illm/README.md
projects/illm/default_transforms.py
projects/illm/eval_folder_example.py
projects/illm/gan_compression_module.py
projects/illm/image_module.py
projects/illm/open_images_datamodule.py
projects/illm/requirements.txt
projects/illm/results.csv
projects/illm/target_compression_module.py
projects/illm/train.py
projects/illm/conf/config.yaml
projects/illm/conf/data/local_data_openimages.yaml
projects/illm/conf/discriminator/condunet_ch1025_factor8_context220.yaml
projects/illm/conf/distortion_loss/mse_distortion.yaml
projects/illm/conf/distortion_loss/mse_lpips.yaml
projects/illm/conf/image_logs/default_logs.yaml
projects/illm/conf/latent_projector/vqvae_xcit_p8_ch64_cb1024_h8.yaml
projects/illm/conf/mode/submitit_multi_node.yaml
projects/illm/conf/model/hific_autoencoder.yaml
projects/illm/conf/optimizer/model_adamw.yaml
projects/illm/conf/optimizer/model_adamw_disc_const.yaml
projects/illm/conf/pretrained_autoencoder/example.yaml
projects/illm/conf/pretrained_autoencoder/nogan_target0.14bpp.yaml
projects/illm/conf/pretrained_latent_autoencoder/20221108_vqvae_xcit_p8_ch64_cb1024_h8.yaml
projects/illm/conf/pretrained_latent_autoencoder/example_local.yaml
projects/illm/conf/rate_target/notarget.yaml
projects/illm/conf/rate_target/target-0.00875.yaml
projects/illm/conf/rate_target/target-0.0175.yaml
projects/illm/conf/rate_target/target-0.035.yaml
projects/illm/conf/rate_target/target-0.07.yaml
projects/illm/conf/rate_target/target-0.14.yaml
projects/illm/conf/rate_target/target-0.3.yaml
projects/illm/conf/rate_target/target-0.45.yaml
projects/illm/conf/rate_target/target-0.9.yaml
projects/illm/conf/test_dataloaders/test.yaml
projects/illm/conf/trainer/default_trainer.yaml
projects/jax_entropy_coders/README.md
projects/jax_entropy_coders/jax_arithmetic_coder.py
projects/jax_entropy_coders/requirements.txt
projects/jax_entropy_coders/test_entropy_coders.py
projects/jax_entropy_coders/craystack/__init__.py
projects/jax_entropy_coders/craystack/_backend.py
projects/jax_entropy_coders/craystack/codecs.py
projects/jax_entropy_coders/craystack/coder.py
projects/scale_hyperprior_lightning/README.md
projects/scale_hyperprior_lightning/requirements.txt
projects/scale_hyperprior_lightning/scale_hyperprior.py
projects/scale_hyperprior_lightning/train.py
projects/scale_hyperprior_lightning/vimeo.py
projects/scale_hyperprior_lightning/config/base.yaml
projects/scale_hyperprior_lightning/config/mode/submitit_multi_node.yaml
projects/scale_hyperprior_lightning/config/mode/submitit_single_node.yaml
projects/scale_hyperprior_lightning/tests/test_scale_hyperprior_lightning.py
projects/torch_vct/.gitignore
projects/torch_vct/README.md
projects/torch_vct/__init__.py
projects/torch_vct/environment.yml
projects/torch_vct/model_lightning.py
projects/torch_vct/model_pipeline.py
projects/torch_vct/model_train.py
projects/torch_vct/config/train_config.yaml
projects/torch_vct/config/analysis_transform/elic.yaml
projects/torch_vct/config/analysis_transform/gdn.yaml
projects/torch_vct/config/datamodule/kinetics.yaml
projects/torch_vct/config/datamodule/vimeo.yaml
projects/torch_vct/config/mode/submitit_multi_node.yaml
projects/torch_vct/config/optim/adam.yaml
projects/torch_vct/config/optim/adamw.yaml
projects/torch_vct/config/optim/lion.yaml
projects/torch_vct/config/scheduler/cosine.yaml
projects/torch_vct/config/scheduler/exponential.yaml
projects/torch_vct/config/scheduler/linearrampcosine.yaml
projects/torch_vct/config/scheduler/linearramplinear.yaml
projects/torch_vct/config/scheduler/linearwarmupcosine.yaml
projects/torch_vct/config/synthesis_transform/elic.yaml
projects/torch_vct/config/synthesis_transform/gdn.yaml
projects/torch_vct/config/test_datamodule/uvg.yaml
projects/torch_vct/datamodules/__init__.py
projects/torch_vct/datamodules/kinetics.py
projects/torch_vct/datamodules/uvg.py
projects/torch_vct/datamodules/video_data_api.py
projects/torch_vct/datamodules/vimeo.py
projects/torch_vct/fig/vct_rd.png
projects/torch_vct/neural/__init__.py
projects/torch_vct/neural/bottlenecks.py
projects/torch_vct/neural/entropy_model.py
projects/torch_vct/neural/entropy_model_layers.py
projects/torch_vct/neural/layers_utils.py
projects/torch_vct/neural/patcher.py
projects/torch_vct/neural/transforms.py
projects/torch_vct/tests/__init__.py
projects/torch_vct/tests/neural/__init__.py
projects/torch_vct/tests/neural/test_entropy_model_layers.py
projects/torch_vct/tests/neural/test_layers_utils.py
projects/torch_vct/tests/neural/test_patcher.py
projects/torch_vct/utils/__init__.py
projects/torch_vct/utils/checkpoint_loaders.py
projects/torch_vct/utils/datavis.py
projects/torch_vct/utils/hydra_tools.py
projects/torch_vct/utils/memoize.py
projects/torch_vct/utils/schedulers.py
tests/conftest.py
tests/test_layers_gdn.py
tests/test_models.py
tests/utils.py
tests/cached_data/create_cached_tensorflow_data.py
tests/cached_data/dense_image_warp.pkl
tests/data/__init__.py
tests/data/test_clic_2020_image.py
tests/data/test_clic_2020_video.py
tests/data/test_div2k.py
tests/data/test_kodak.py
tests/data/test_open_images.py
tests/data/test_vimeo_90k_septuplet.py
tests/distributions/__init__.py
tests/distributions/test_noisy_normal.py
tests/distributions/test_uniform_noise.py
tests/functional/__init__.py
tests/functional/test_count_flops.py
tests/functional/test_dense_image_warp.py
tests/functional/test_estimate_tails.py
tests/functional/test_hsv_to_rgb.py
tests/functional/test_image_255_scale.py
tests/functional/test_information_content.py
tests/functional/test_log_cdf.py
tests/functional/test_log_expm1.py
tests/functional/test_log_ndtr.py
tests/functional/test_log_survival_function.py
tests/functional/test_lower_bound.py
tests/functional/test_lower_tail.py
tests/functional/test_ndtr.py
tests/functional/test_optical_flow_to_rgb.py
tests/functional/test_pad_image.py
tests/functional/test_quantization_offset.py
tests/functional/test_soft_round.py
tests/functional/test_soft_round_conditional_mean.py
tests/functional/test_soft_round_inverse.py
tests/functional/test_survival_function.py
tests/functional/test_upper_tail.py
tests/layers/__init__.py
tests/layers/test_channelnorm.py
tests/layers/test_continuous_entropy.py
tests/loss_fn/test_binary_cross_entropy.py
tests/loss_fn/test_mse_loss.py
tests/loss_fn/test_mse_lpips_loss.py
tests/loss_fn/test_oasis_loss.py
tests/metrics/__init__.py
tests/metrics/test_calc_psnr.py
tests/metrics/test_dists.py
tests/metrics/test_fid.py
tests/metrics/test_kid.py
tests/metrics/test_msssim.py
tests/metrics/test_pickle_size.py
tests/metrics/test_swav_fid.py
tests/metrics/test_update_patch_fid.py
tests/models/__init__.py
tests/models/test_hific_autoencoder.py
tests/models/test_hific_encoder.py
tests/models/test_hific_generator.py
tests/models/test_unet_discriminator.py
tutorials/Flop_Count_Example.ipynb
tutorials/Metrics_Example.ipynb
tutorials/image_compression.ipynb