.gitattributes
.gitignore
.pre-commit-config.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/entropy_coders/__init__.py
neuralcompression/entropy_coders/jax_arithemetic_coder.py
neuralcompression/entropy_coders/craystack/__init__.py
neuralcompression/entropy_coders/craystack/_backend.py
neuralcompression/entropy_coders/craystack/codecs.py
neuralcompression/entropy_coders/craystack/coder.py
neuralcompression/ext/__init__.py
neuralcompression/ext/_extension_loader.py
neuralcompression/ext/pmf_to_quantized_cdf_py.cc
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/_pmf_to_quantized_cdf.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/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/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/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_entropy_coders.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_pmf_to_quantized_cdf.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