LICENSE
README.md
pyproject.toml
src/spherinator/__init__.py
src/spherinator/main.py
src/spherinator.egg-info/PKG-INFO
src/spherinator.egg-info/SOURCES.txt
src/spherinator.egg-info/dependency_links.txt
src/spherinator.egg-info/entry_points.txt
src/spherinator.egg-info/requires.txt
src/spherinator.egg-info/top_level.txt
src/spherinator/callbacks/__init__.py
src/spherinator/callbacks/log_reconstruction_callback.py
src/spherinator/data/__init__.py
src/spherinator/data/data_module.py
src/spherinator/data/images_data_module.py
src/spherinator/data/images_dataset.py
src/spherinator/data/mnist_data_module.py
src/spherinator/data/parquet_data_module.py
src/spherinator/data/parquet_dataset.py
src/spherinator/data/parquet_dataset_sampling.py
src/spherinator/data/parquet_dataset_with_error.py
src/spherinator/data/parquet_iterable_data_module.py
src/spherinator/data/parquet_iterable_dataset.py
src/spherinator/data/preprocessing.py
src/spherinator/distributions/__init__.py
src/spherinator/distributions/power_spherical.py
src/spherinator/distributions/truncated_normal.py
src/spherinator/distributions/truncated_normal_distribution.py
src/spherinator/models/__init__.py
src/spherinator/models/autoencoder.py
src/spherinator/models/consecutive_conv_1d_layers.py
src/spherinator/models/consecutive_conv_2d_layers.py
src/spherinator/models/consecutive_conv_transpose_1d_layers.py
src/spherinator/models/consecutive_conv_transpose_2d_layers.py
src/spherinator/models/convolutional_decoder_1d.py
src/spherinator/models/convolutional_decoder_2d.py
src/spherinator/models/convolutional_encoder_1d.py
src/spherinator/models/convolutional_encoder_2d.py
src/spherinator/models/embedding_reconstruction.py
src/spherinator/models/export_onnx.py
src/spherinator/models/huggingface_vit_encoder.py
src/spherinator/models/mlp.py
src/spherinator/models/upsampling_decoder_2d.py
src/spherinator/models/variational_autoencoder.py
src/spherinator/models/weights_provider.py
src/spherinator/models/yaml2model.py
tests/test_autoencoder.py
tests/test_cnn_1d.py
tests/test_cnn_2d.py
tests/test_images_data_module.py
tests/test_images_dataset.py
tests/test_log_reconstruction_callback.py
tests/test_mlp.py
tests/test_mnist_data_module.py
tests/test_onnx_export.py
tests/test_parquet_dataset.py
tests/test_parquet_iterable_dataset.py
tests/test_power_spherical.py
tests/test_resnet.py
tests/test_torch.py
tests/test_variational_autoencoder.py