LICENSE
README.md
pyproject.toml
lightning_uq_box/__init__.py
lightning_uq_box/__main__.py
lightning_uq_box/main.py
lightning_uq_box/py.typed
lightning_uq_box.egg-info/PKG-INFO
lightning_uq_box.egg-info/SOURCES.txt
lightning_uq_box.egg-info/dependency_links.txt
lightning_uq_box.egg-info/entry_points.txt
lightning_uq_box.egg-info/requires.txt
lightning_uq_box.egg-info/top_level.txt
lightning_uq_box/datamodules/__init__.py
lightning_uq_box/datamodules/toy_8_gaussians.py
lightning_uq_box/datamodules/toy_donut.py
lightning_uq_box/datamodules/toy_due.py
lightning_uq_box/datamodules/toy_gaussian_sideways.py
lightning_uq_box/datamodules/toy_half_moons.py
lightning_uq_box/datamodules/toy_heteroscedastic.py
lightning_uq_box/datamodules/toy_image_classification.py
lightning_uq_box/datamodules/toy_image_regression.py
lightning_uq_box/datamodules/toy_image_segmentation.py
lightning_uq_box/datamodules/toy_pixelwise_regression.py
lightning_uq_box/datamodules/toy_sine.py
lightning_uq_box/datamodules/utils.py
lightning_uq_box/datasets/__init__.py
lightning_uq_box/datasets/toy_8_gaussians.py
lightning_uq_box/datasets/toy_donut.py
lightning_uq_box/datasets/toy_gaussian_sideways.py
lightning_uq_box/datasets/toy_image_classification.py
lightning_uq_box/datasets/toy_image_regression.py
lightning_uq_box/datasets/toy_image_segmentation.py
lightning_uq_box/datasets/toy_pixelwise_regression.py
lightning_uq_box/eval_utils/__init__.py
lightning_uq_box/eval_utils/uq_computation.py
lightning_uq_box/models/__init__.py
lightning_uq_box/models/cards.py
lightning_uq_box/models/density_layers.py
lightning_uq_box/models/fc_resnet.py
lightning_uq_box/models/hierarchical_prob_unet.py
lightning_uq_box/models/masked_conv.py
lightning_uq_box/models/mixture_density.py
lightning_uq_box/models/mlp.py
lightning_uq_box/models/prob_unet.py
lightning_uq_box/models/vae.py
lightning_uq_box/models/bnn_layers/__init__.py
lightning_uq_box/models/bnn_layers/base_variational.py
lightning_uq_box/models/bnn_layers/bnn_utils.py
lightning_uq_box/models/bnn_layers/conv_variational.py
lightning_uq_box/models/bnn_layers/linear_variational.py
lightning_uq_box/models/bnn_layers/rnn_layer.py
lightning_uq_box/models/bnnlv/latent_variable_network.py
lightning_uq_box/models/bnnlv/utils.py
lightning_uq_box/models/masked_ensemble/__init__.py
lightning_uq_box/models/masked_ensemble/masked_layers.py
lightning_uq_box/models/masked_ensemble/utils.py
lightning_uq_box/uq_methods/__init__.py
lightning_uq_box/uq_methods/base.py
lightning_uq_box/uq_methods/bnn_lv_vi.py
lightning_uq_box/uq_methods/bnn_vi.py
lightning_uq_box/uq_methods/bnn_vi_elbo.py
lightning_uq_box/uq_methods/cards.py
lightning_uq_box/uq_methods/conformal_qr.py
lightning_uq_box/uq_methods/deep_ensemble.py
lightning_uq_box/uq_methods/deep_evidential_regression.py
lightning_uq_box/uq_methods/deep_kernel_learning.py
lightning_uq_box/uq_methods/density_uncertainty.py
lightning_uq_box/uq_methods/deterministic_uncertainty_estimation.py
lightning_uq_box/uq_methods/hierarchical_prob_unet.py
lightning_uq_box/uq_methods/img2img_conformal.py
lightning_uq_box/uq_methods/inference_time_augmentation.py
lightning_uq_box/uq_methods/laplace_model.py
lightning_uq_box/uq_methods/loss_functions.py
lightning_uq_box/uq_methods/masked_ensemble.py
lightning_uq_box/uq_methods/mc_dropout.py
lightning_uq_box/uq_methods/mean_variance_estimation.py
lightning_uq_box/uq_methods/metrics.py
lightning_uq_box/uq_methods/mixture_density.py
lightning_uq_box/uq_methods/prob_unet.py
lightning_uq_box/uq_methods/quantile_regression.py
lightning_uq_box/uq_methods/raps.py
lightning_uq_box/uq_methods/sgld.py
lightning_uq_box/uq_methods/sngp.py
lightning_uq_box/uq_methods/spectral_normalized_layers.py
lightning_uq_box/uq_methods/swag.py
lightning_uq_box/uq_methods/temp_scaling.py
lightning_uq_box/uq_methods/utils.py
lightning_uq_box/uq_methods/vae.py
lightning_uq_box/uq_methods/vbll.py
lightning_uq_box/uq_methods/zigzag.py
lightning_uq_box/viz_utils/__init__.py
lightning_uq_box/viz_utils/visualization_tools.py
tests/test_main.py