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/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/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/film.py
kappamodules/utils/__init__.py
kappamodules/utils/param_checking.py
kappamodules/utils/shapes.py
kappamodules/utils/str_to_ctor.py
kappamodules/vit/__init__.py
kappamodules/vit/vit_batchnorm.py
kappamodules/vit/vit_block.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