LICENSE
README.md
pyproject.toml
src/kerex.egg-info/PKG-INFO
src/kerex.egg-info/SOURCES.txt
src/kerex.egg-info/dependency_links.txt
src/kerex.egg-info/requires.txt
src/kerex.egg-info/top_level.txt
src/kerex/blocks/__init__.py
src/kerex/blocks/attention/__init__.py
src/kerex/blocks/attention/attention.py
src/kerex/blocks/attention/base_attention.py
src/kerex/blocks/autoencoder/__init__.py
src/kerex/blocks/autoencoder/base_decoder.py
src/kerex/blocks/autoencoder/base_encoder.py
src/kerex/blocks/autoencoder/decoder.py
src/kerex/blocks/autoencoder/encoder.py
src/kerex/blocks/tsmixer/__init__.py
src/kerex/blocks/tsmixer/tsmixer.py
src/kerex/callbacks/__init__.py
src/kerex/callbacks/callbacks.py
src/kerex/dataloader/__init__.py
src/kerex/dataloader/dataloader.py
src/kerex/dataloader/augmentation/__init__.py
src/kerex/dataloader/augmentation/random_noise.py
src/kerex/dataloader/augmentation/random_rotation.py
src/kerex/layers/__init__.py
src/kerex/layers/conv/__init__.py
src/kerex/layers/conv/base_conv.py
src/kerex/layers/conv/conv1d.py
src/kerex/layers/conv/conv2d.py
src/kerex/layers/conv/conv3d.py
src/kerex/layers/fno/__init__.py
src/kerex/layers/fno/base_fno.py
src/kerex/layers/fno/fno.py
src/kerex/layers/fno/spectral_conv/__init__.py
src/kerex/layers/fno/spectral_conv/base_spectral_conv.py
src/kerex/layers/fno/spectral_conv/spectral_conv.py
src/kerex/layers/merge/__init__.py
src/kerex/layers/merge/attention_gate.py
src/kerex/layers/wrapper/__init__.py
src/kerex/layers/wrapper/attention.py
src/kerex/layers/wrapper/cbam.py
src/kerex/layers/wrapper/film.py
src/kerex/layers/wrapper/residual.py
src/kerex/layers/wrapper/temporal_slice.py
src/kerex/layers/wrapper/weight_normalization.py
src/kerex/models/__init__.py
src/kerex/models/autoencoder/__init__.py
src/kerex/models/autoencoder/base_models.py
src/kerex/models/autoencoder/fully_convolutional_network.py
src/kerex/models/autoencoder/unet.py
src/kerex/models/neural_operator/__init__.py
src/kerex/models/neural_operator/base_neural_operator.py
src/kerex/models/neural_operator/neural_operator.py
src/kerex/models/tsmixer/__init__.py
src/kerex/models/tsmixer/tsmixer.py
src/kerex/ops/__init__.py
src/kerex/ops/helper.py
src/kerex/ops/fft/__init__.py
src/kerex/ops/fft/jax.py
src/kerex/ops/fft/tensorflow.py
src/kerex/optimizers/__init__.py
src/kerex/optimizers/adamw.py