.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
LICENSE
MANIFEST.in
NOTICE
README.md
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
docs/Makefile
docs/make.bat
docs/examples/README.rst
docs/examples/image.png
docs/examples/plot_image_compression.py
docs/source/conf.py
docs/source/data.rst
docs/source/distributions.rst
docs/source/entropy_coders.rst
docs/source/functional.rst
docs/source/index.rst
docs/source/layers.rst
docs/source/metrics.rst
docs/source/models.rst
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/data/__init__.py
neuralcompression/data/_clic_2020_image.py
neuralcompression/data/_clic_2020_video.py
neuralcompression/data/_kodak.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/_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/_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/_analysis_transformation_2d.py
neuralcompression/layers/_continuous_entropy.py
neuralcompression/layers/_generalized_divisive_normalization.py
neuralcompression/layers/_hyper_analysis_transformation_2d.py
neuralcompression/layers/_hyper_synthesis_transformation_2d.py
neuralcompression/layers/_non_negative_parameterization.py
neuralcompression/layers/_rate_mse_distortion_loss.py
neuralcompression/layers/_synthesis_transformation_2d.py
neuralcompression/layers/gdn.py
neuralcompression/metrics/__init__.py
neuralcompression/metrics/distortion.py
neuralcompression/models/__init__.py
neuralcompression/models/_factorized_prior_autoencoder.py
neuralcompression/models/_hyperprior_autoencoder.py
neuralcompression/models/_mean_scale_hyperprior_autoencoder.py
neuralcompression/models/_prior_autoencoder.py
neuralcompression/models/_scale_hyperprior_autoencoder.py
neuralcompression/models/deep_video_compression.py
neuralcompression/models/scale_hyperprior.py
neuralcompression/models/_hific/__init__.py
neuralcompression/models/_hific/_channel_norm_2d.py
neuralcompression/models/_hific/_hific_discriminator.py
neuralcompression/models/_hific/_hific_encoder.py
neuralcompression/models/_hific/_hific_generator.py
neuralcompression/models/_hific/_least_squares_adversarial_loss.py
neuralcompression/models/_hific/_non_saturating_adversarial_loss.py
neuralcompression/models/_hific/_weighted_rate_loss.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/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
projects/variational_image_compression/lightning/__init__.py
projects/variational_image_compression/lightning/_factorized_prior_autoencoder.py
projects/variational_image_compression/lightning/_mean_scale_hyperprior_autoencoder.py
projects/variational_image_compression/lightning/_prior_autoencoder.py
projects/variational_image_compression/lightning/_scale_hyperprior_autoencoder.py
projects/variational_image_compression/test/__init__.py
projects/variational_image_compression/test/test_factorized_prior_autoencoder.py
projects/variational_image_compression/test/test_mean_scale_hyperprior_autoencoder.py
projects/variational_image_compression/test/test_scale_hyperprior_autoencoder.py
tests/test_layers_gdn.py
tests/test_models.py
tests/test_scale_hyperprior.py
tests/utils.py
tests/data/__init__.py
tests/data/test_clic_2020_image.py
tests/data/test_clic_2020_video.py
tests/data/test_kodak.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_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_lpips.py
tests/functional/test_multiscale_structural_similarity.py
tests/functional/test_ndtr.py
tests/functional/test_optical_flow_to_rgb.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_analysis_transformation_2d.py
tests/layers/test_continuous_entropy.py
tests/layers/test_generalized_divisive_normalization.py
tests/layers/test_hyper_analysis_transformation_2d.py
tests/layers/test_hyper_synthesis_transformation_2d.py
tests/layers/test_non_negative_parameterization.py
tests/layers/test_rate_mse_distortion_loss.py
tests/layers/test_synthesis_transformation_2d.py
tests/models/__init__.py
tests/models/test_factorized_prior_autoencoder.py
tests/models/test_mean_scale_hyperprior_autoencoder.py
tests/models/test_prior_autoencoder.py
tests/models/test_scale_hyperprior_autoencoder.py
tests/models/_hific/test_hific_discriminator.py
tests/models/_hific/test_hific_encoder.py
tests/models/_hific/test_hific_generator.py
tutorials/Flop_Count_Example.ipynb
tutorials/Metrics_Example.ipynb
tutorials/image_compression.ipynb