LICENSE
README.md
pyproject.toml
setup.cfg
kappamodules/__init__.py
kappamodules.egg-info/PKG-INFO
kappamodules.egg-info/SOURCES.txt
kappamodules.egg-info/dependency_links.txt
kappamodules.egg-info/requires.txt
kappamodules.egg-info/top_level.txt
kappamodules/attention/__init__.py
kappamodules/attention/dot_product_attention.py
kappamodules/attention/dot_product_attention_slow.py
kappamodules/attention/linear_attention.py
kappamodules/attention/perceiver_attention.py
kappamodules/diffaug/__init__.py
kappamodules/diffaug/gaussian_blur.py
kappamodules/functional/__init__.py
kappamodules/functional/patchify.py
kappamodules/functional/pos_embed.py
kappamodules/init/__init__.py
kappamodules/init/functional.py
kappamodules/layers/__init__.py
kappamodules/layers/drop_path.py
kappamodules/layers/identity.py
kappamodules/layers/normalize.py
kappamodules/layers/paramless_batchnorm.py
kappamodules/layers/residual.py
kappamodules/layers/weight_norm_linear.py
kappamodules/losses/__init__.py
kappamodules/losses/l1_loss.py
kappamodules/losses/mse_loss.py
kappamodules/modulation/__init__.py
kappamodules/modulation/dit.py
kappamodules/modulation/film.py
kappamodules/modulation/functional.py
kappamodules/modulation/timestep_embed.py
kappamodules/ssl/__init__.py
kappamodules/ssl/contrastive_predictor.py
kappamodules/ssl/contrastive_projector.py
kappamodules/ssl/dino_projector.py
kappamodules/transformer/__init__.py
kappamodules/transformer/mlp.py
kappamodules/transformer/perceiver_block.py
kappamodules/transformer/postnorm_block.py
kappamodules/utils/__init__.py
kappamodules/utils/mode_to_ctor.py
kappamodules/utils/param_checking.py
kappamodules/utils/shapes.py
kappamodules/utils/tensor_cache.py
kappamodules/vit/__init__.py
kappamodules/vit/dit_block.py
kappamodules/vit/vit_batchnorm.py
kappamodules/vit/vit_block.py
kappamodules/vit/vit_blocks.py
kappamodules/vit/vit_class_tokens.py
kappamodules/vit/vit_mlp.py
kappamodules/vit/vit_patch_embed.py
kappamodules/vit/vit_pos_embed.py
kappamodules/vit/vit_seperate_norm.py