.flake8
.gitignore
.pre-commit-config.yaml
CONTRIBUTING.md
LICENSE.txt
MANIFEST.in
README.md
SPDX.spdx
pyproject.toml
.github/dependabot.yml
.github/workflows/nightly.yml
.github/workflows/publish_docs.yml
.github/workflows/publish_rai_experiments.yml
.github/workflows/pypi_publish.yml
.github/workflows/test_docs.yml
.github/workflows/tox_run.yml
brand/logo_no_text.png
brand/logo_no_text_small.png
deps/requirements-pyright.txt
docs/Makefile
docs/README.md
docs/conda_env.yml
docs/make.bat
docs/requirements.txt
docs/source/api_reference.rst
docs/source/changes.rst
docs/source/conf.py
docs/source/explanation.rst
docs/source/how_tos.rst
docs/source/index.rst
docs/source/ref_augmentations.rst
docs/source/ref_datasets.rst
docs/source/ref_losses.rst
docs/source/ref_mushin.rst
docs/source/ref_optim.rst
docs/source/ref_perturbation.rst
docs/source/ref_utils.rst
docs/source/tutorials.rst
docs/source/_static/apple-touch-icon.png
docs/source/_static/favicon-16x16.png
docs/source/_static/favicon-32x32.png
docs/source/explanation/perturbations.rst
docs/source/generated/rai_toolbox.augmentations.augmix.AugMix.rst
docs/source/generated/rai_toolbox.augmentations.augmix.Fork.rst
docs/source/generated/rai_toolbox.augmentations.augmix.augment_and_mix.rst
docs/source/generated/rai_toolbox.augmentations.fourier.FourierBasis.rst
docs/source/generated/rai_toolbox.augmentations.fourier.create_heatmaps.rst
docs/source/generated/rai_toolbox.augmentations.fourier.generate_fourier_bases.rst
docs/source/generated/rai_toolbox.datasets.CIFAR100C.rst
docs/source/generated/rai_toolbox.datasets.CIFAR10C.rst
docs/source/generated/rai_toolbox.datasets.CIFAR10P1.rst
docs/source/generated/rai_toolbox.datasets.ImageNet.rst
docs/source/generated/rai_toolbox.datasets.ImageNetM10.rst
docs/source/generated/rai_toolbox.datasets.RestrictedImageNet.rst
docs/source/generated/rai_toolbox.evaluating.rst
docs/source/generated/rai_toolbox.freeze.rst
docs/source/generated/rai_toolbox.frozen.rst
docs/source/generated/rai_toolbox.losses.jensen_shannon_divergence.rst
docs/source/generated/rai_toolbox.mushin.BaseWorkflow.rst
docs/source/generated/rai_toolbox.mushin.HydraDDP.rst
docs/source/generated/rai_toolbox.mushin.MetricsCallback.rst
docs/source/generated/rai_toolbox.mushin.MultiRunMetricsWorkflow.rst
docs/source/generated/rai_toolbox.mushin.RobustnessCurve.rst
docs/source/generated/rai_toolbox.negate.rst
docs/source/generated/rai_toolbox.optim.ChainedParamTransformingOptimizer.rst
docs/source/generated/rai_toolbox.optim.ClampedGradientOptimizer.rst
docs/source/generated/rai_toolbox.optim.ClampedParameterOptimizer.rst
docs/source/generated/rai_toolbox.optim.FrankWolfe.rst
docs/source/generated/rai_toolbox.optim.L1FrankWolfe.rst
docs/source/generated/rai_toolbox.optim.L1NormedGradientOptim.rst
docs/source/generated/rai_toolbox.optim.L1qFrankWolfe.rst
docs/source/generated/rai_toolbox.optim.L1qNormedGradientOptim.rst
docs/source/generated/rai_toolbox.optim.L2FrankWolfe.rst
docs/source/generated/rai_toolbox.optim.L2NormedGradientOptim.rst
docs/source/generated/rai_toolbox.optim.L2ProjectedOptim.rst
docs/source/generated/rai_toolbox.optim.LinfFrankWolfe.rst
docs/source/generated/rai_toolbox.optim.LinfProjectedOptim.rst
docs/source/generated/rai_toolbox.optim.ParamTransformingOptimizer.rst
docs/source/generated/rai_toolbox.optim.SignedGradientOptim.rst
docs/source/generated/rai_toolbox.optim.TopQGradientOptimizer.rst
docs/source/generated/rai_toolbox.perturbations.AdditivePerturbation.rst
docs/source/generated/rai_toolbox.perturbations.PerturbationModel.rst
docs/source/generated/rai_toolbox.perturbations.gradient_ascent.rst
docs/source/generated/rai_toolbox.perturbations.random_restart.rst
docs/source/generated/rai_toolbox.perturbations.uniform_like_l1_n_ball_.rst
docs/source/generated/rai_toolbox.perturbations.uniform_like_l2_n_ball_.rst
docs/source/generated/rai_toolbox.perturbations.uniform_like_linf_n_ball_.rst
docs/source/generated/rai_toolbox.to_batch.rst
docs/source/how_to/custom_optim.rst
docs/source/how_to/deterministic_workflow.rst
docs/source/how_to/hydraddp.rst
docs/source/how_to/univ_adv_pert.rst
docs/source/tutorials/Building-Workflows.ipynb
docs/source/tutorials/CIFAR10-Adversarial-Perturbations.ipynb
docs/source/tutorials/ImageNet-Concept-Probing.ipynb
examples/CIFAR10-Adversarial-Perturbations.ipynb
examples/ImageNet-Concept-Probing.ipynb
examples/MNIST-Translation-Robustness.ipynb
examples/fourier/FourierBasisPerturbations.ipynb
examples/fourier/utils.py
experiments/LICENSE.txt
experiments/README.md
experiments/setup.cfg
experiments/setup.py
experiments/adversarial_training/README.md
experiments/adversarial_training/TestRobustness.ipynb
experiments/adversarial_training/configs.py
experiments/adversarial_training/solver.py
experiments/adversarial_training/test_experiment.py
experiments/adversarial_training/train.py
experiments/benchmarking/benchmarks.ipynb
experiments/madry/README.md
experiments/madry/madry.py
experiments/madry/test_experiment.py
experiments/src/rai_experiments/__init__.py
experiments/src/rai_experiments/models/__init__.py
experiments/src/rai_experiments/models/_resnet_blocks.py
experiments/src/rai_experiments/models/pretrained.py
experiments/src/rai_experiments/models/resnet.py
experiments/src/rai_experiments/models/small_resnet.py
experiments/src/rai_experiments/utils/__init__.py
experiments/src/rai_experiments/utils/imagenet_labels.py
experiments/src/rai_experiments/utils/visualizations.py
experiments/tests/test_rai_experiments.py
experiments/universal_perturbation/README.md
experiments/universal_perturbation/UniversalPerturbation.ipynb
experiments/universal_perturbation/configs.py
experiments/universal_perturbation/solver.py
experiments/universal_perturbation/test_experiment.py
experiments/universal_perturbation/train.py
experiments/xai/ConceptProbing.ipynb
experiments/xai/Perturbations.ipynb
experiments/xai/README.md
src/rai_toolbox/__init__.py
src/rai_toolbox/_typing.py
src/rai_toolbox/_version.py
src/rai_toolbox/py.typed
src/rai_toolbox.egg-info/PKG-INFO
src/rai_toolbox.egg-info/SOURCES.txt
src/rai_toolbox.egg-info/dependency_links.txt
src/rai_toolbox.egg-info/requires.txt
src/rai_toolbox.egg-info/top_level.txt
src/rai_toolbox/_utils/__init__.py
src/rai_toolbox/_utils/itertools.py
src/rai_toolbox/_utils/stateful.py
src/rai_toolbox/_utils/tqdm.py
src/rai_toolbox/augmentations/__init__.py
src/rai_toolbox/augmentations/augmix/__init__.py
src/rai_toolbox/augmentations/augmix/_implementation.py
src/rai_toolbox/augmentations/augmix/transforms.py
src/rai_toolbox/augmentations/fourier/__init__.py
src/rai_toolbox/augmentations/fourier/_fourier_basis.py
src/rai_toolbox/augmentations/fourier/_implementations.py
src/rai_toolbox/augmentations/fourier/transforms.py
src/rai_toolbox/datasets/__init__.py
src/rai_toolbox/datasets/_cifar10_base.py
src/rai_toolbox/datasets/_imagenet_base.py
src/rai_toolbox/datasets/_utils.py
src/rai_toolbox/datasets/cifar10_extensions.py
src/rai_toolbox/datasets/cifar_corruptions.py
src/rai_toolbox/losses/__init__.py
src/rai_toolbox/losses/_jensen_shannon_divergence.py
src/rai_toolbox/losses/_utils.py
src/rai_toolbox/mushin/__init__.py
src/rai_toolbox/mushin/_compatibility.py
src/rai_toolbox/mushin/_utils.py
src/rai_toolbox/mushin/workflows.py
src/rai_toolbox/mushin/lightning/__init__.py
src/rai_toolbox/mushin/lightning/_pl_main.py
src/rai_toolbox/mushin/lightning/callbacks.py
src/rai_toolbox/mushin/lightning/launchers.py
src/rai_toolbox/mushin/testing/__init__.py
src/rai_toolbox/mushin/testing/lightning.py
src/rai_toolbox/optim/__init__.py
src/rai_toolbox/optim/frank_wolfe.py
src/rai_toolbox/optim/lp_space.py
src/rai_toolbox/optim/misc.py
src/rai_toolbox/optim/optimizer.py
src/rai_toolbox/perturbations/__init__.py
src/rai_toolbox/perturbations/init.py
src/rai_toolbox/perturbations/models.py
src/rai_toolbox/perturbations/solvers.py