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/architectures/__init__.py
kappamodules/architectures/upt/__init__.py
kappamodules/architectures/upt/perceiver_decoder.py
kappamodules/attention/__init__.py
kappamodules/attention/dot_product_attention.py
kappamodules/attention/dot_product_attention_slow.py
kappamodules/attention/efficient_attention.py
kappamodules/attention/linformer_attention.py
kappamodules/attention/localgrid_attention.py
kappamodules/attention/mmdit_dot_product_attention.py
kappamodules/attention/mmmdit_dot_product_attention.py
kappamodules/attention/perceiver_attention.py
kappamodules/attention/transsolver_attention.py
kappamodules/convolution/__init__.py
kappamodules/convolution/convnext.py
kappamodules/diffaug/__init__.py
kappamodules/diffaug/gaussian_blur.py
kappamodules/functional/__init__.py
kappamodules/functional/logscale.py
kappamodules/functional/patchify.py
kappamodules/functional/pos_embed.py
kappamodules/init/__init__.py
kappamodules/init/functional.py
kappamodules/layers/__init__.py
kappamodules/layers/async_batchnorm.py
kappamodules/layers/continuous_pos_embed.py
kappamodules/layers/continuous_sincos_embed.py
kappamodules/layers/datanorm.py
kappamodules/layers/drop_path.py
kappamodules/layers/identity.py
kappamodules/layers/ignore_args_and_kwargs_wrapper.py
kappamodules/layers/layer_scale.py
kappamodules/layers/layernorm.py
kappamodules/layers/learned_batchnorm.py
kappamodules/layers/linear_projection.py
kappamodules/layers/normalize.py
kappamodules/layers/paramless_batchnorm.py
kappamodules/layers/regular_grid_sincos_embed.py
kappamodules/layers/residual.py
kappamodules/layers/rms_norm.py
kappamodules/layers/sequential.py
kappamodules/layers/weight_norm_linear.py
kappamodules/losses/__init__.py
kappamodules/losses/l1_loss.py
kappamodules/losses/mse_loss.py
kappamodules/mlp/__init__.py
kappamodules/mlp/mlp.py
kappamodules/mlp/prenorm_mlp_block.py
kappamodules/mlp/swiglu_mlp.py
kappamodules/modulation/__init__.py
kappamodules/modulation/dit.py
kappamodules/modulation/film.py
kappamodules/modulation/functional.py
kappamodules/modulation/timestep_embed.py
kappamodules/norm/__init__.py
kappamodules/norm/global_response_norm.py
kappamodules/pooling/__init__.py
kappamodules/pooling/cross_attention_pooling.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/dit_block.py
kappamodules/transformer/dit_perceiver_block.py
kappamodules/transformer/dit_perceiver_pooling_block.py
kappamodules/transformer/mlp.py
kappamodules/transformer/mmdit_block.py
kappamodules/transformer/mmmdit_block.py
kappamodules/transformer/perceiver_block.py
kappamodules/transformer/perceiver_pooling_block.py
kappamodules/transformer/postnorm_block.py
kappamodules/transformer/prenorm_block.py
kappamodules/unet/__init__.py
kappamodules/unet/unet_denoising_diffusion.py
kappamodules/unet/unet_gino.py
kappamodules/unet/unet_pdearena.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