path	# lines of code
projects/variational_image_compression/lightning/_prior_autoencoder.py	127
projects/variational_image_compression/lightning/__init__.py	4
projects/variational_image_compression/lightning/_factorized_prior_autoencoder.py	49
projects/variational_image_compression/lightning/_mean_scale_hyperprior_autoencoder.py	48
projects/variational_image_compression/lightning/_scale_hyperprior_autoencoder.py	48
projects/deep_video_compression/data_module.py	66
projects/deep_video_compression/config/mode/submitit_multi_node.yaml	9
projects/deep_video_compression/config/mode/submitit_single_node.yaml	6
projects/deep_video_compression/config/training_stages/s1_motion_estimation.yaml	9
projects/deep_video_compression/config/training_stages/s2_motion_compression.yaml	8
projects/deep_video_compression/config/training_stages/s5_total.yaml	11
projects/deep_video_compression/config/training_stages/s3_motion_compensation.yaml	8
projects/deep_video_compression/config/training_stages/s4_total_2frame.yaml	8
projects/deep_video_compression/config/base.yaml	36
projects/deep_video_compression/train.py	120
projects/deep_video_compression/dvc_module.py	240
projects/deep_video_compression/_utils.py	231
projects/scale_hyperprior_lightning/config/mode/submitit_multi_node.yaml	9
projects/scale_hyperprior_lightning/config/mode/submitit_single_node.yaml	6
projects/scale_hyperprior_lightning/config/base.yaml	34
projects/scale_hyperprior_lightning/vimeo.py	55
projects/scale_hyperprior_lightning/train.py	41
projects/scale_hyperprior_lightning/scale_hyperprior.py	110
pyproject.toml	11
MANIFEST.in	5
setup.py	2
neuralcompression/metrics/distortion.py	94
neuralcompression/metrics/__init__.py	4
neuralcompression/layers/gdn.py	43
neuralcompression/layers/_hyper_analysis_transformation_2d.py	43
neuralcompression/layers/_generalized_divisive_normalization.py	117
neuralcompression/layers/__init__.py	9
neuralcompression/layers/_synthesis_transformation_2d.py	50
neuralcompression/layers/_rate_mse_distortion_loss.py	34
neuralcompression/layers/_non_negative_parameterization.py	44
neuralcompression/layers/_analysis_transformation_2d.py	46
neuralcompression/layers/_continuous_entropy.py	225
neuralcompression/layers/_hyper_synthesis_transformation_2d.py	44
neuralcompression/entropy_coders/jax_arithemetic_coder.py	386
neuralcompression/entropy_coders/craystack/_backend.py	155
neuralcompression/entropy_coders/craystack/codecs.py	125
neuralcompression/entropy_coders/craystack/__init__.py	17
neuralcompression/entropy_coders/craystack/coder.py	130
neuralcompression/entropy_coders/__init__.py	2
neuralcompression/ext/__init__.py	9
neuralcompression/ext/_extension_loader.py	9
neuralcompression/ext/pmf_to_quantized_cdf_py.cc	56
neuralcompression/__init__.py	6
neuralcompression/models/_prior_autoencoder.py	101
neuralcompression/models/_hific/_hific_encoder.py	51
neuralcompression/models/_hific/_least_squares_adversarial_loss.py	13
neuralcompression/models/_hific/_non_saturating_adversarial_loss.py	17
neuralcompression/models/_hific/_hific_generator.py	101
neuralcompression/models/_hific/__init__.py	3
neuralcompression/models/_hific/_weighted_rate_loss.py	29
neuralcompression/models/_hific/_hific_discriminator.py	46
neuralcompression/models/_hific/_channel_norm_2d.py	24
neuralcompression/models/__init__.py	8
neuralcompression/models/_factorized_prior_autoencoder.py	47
neuralcompression/models/deep_video_compression.py	422
neuralcompression/models/_mean_scale_hyperprior_autoencoder.py	112
neuralcompression/models/_scale_hyperprior_autoencoder.py	64
neuralcompression/models/_hyperprior_autoencoder.py	101
neuralcompression/models/scale_hyperprior.py	290
neuralcompression/data/_vimeo_90k_septuplet.py	74
neuralcompression/data/_clic_2020_video.py	134
neuralcompression/data/_clic_2020_image.py	75
neuralcompression/data/__init__.py	4
neuralcompression/data/_kodak.py	67
neuralcompression/functional/_hsv_to_rgb.py	17
neuralcompression/functional/_ndtr.py	15
neuralcompression/functional/_log_expm1.py	11
neuralcompression/functional/_estimate_tails.py	30
neuralcompression/functional/_pmf_to_quantized_cdf.py	15
neuralcompression/functional/_information_content.py	18
neuralcompression/functional/__init__.py	22
neuralcompression/functional/_soft_round_inverse.py	11
neuralcompression/functional/_lower_bound.py	44
neuralcompression/functional/_optical_flow_to_color.py	23
neuralcompression/functional/_upper_tail.py	20
neuralcompression/functional/_lpips.py	43
neuralcompression/functional/_count_flops.py	66
neuralcompression/functional/_multiscale_structural_similarity.py	131
neuralcompression/functional/_quantization_offset.py	10
neuralcompression/functional/_log_ndtr.py	34
neuralcompression/functional/_dense_image_warp.py	42
neuralcompression/functional/_survival_function.py	12
neuralcompression/functional/_soft_round_conditional_mean.py	4
neuralcompression/functional/_log_survival_function.py	12
neuralcompression/functional/_log_cdf.py	14
neuralcompression/functional/_soft_round.py	7
neuralcompression/functional/_lower_tail.py	20
neuralcompression/distributions/_uniform_noise.py	81
neuralcompression/distributions/__init__.py	2
neuralcompression/distributions/_noisy_normal.py	13
setup.cfg	105