├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── example.py ├── requirements.txt ├── setup.py └── spectral_ssm ├── __init__.py ├── cifar10.py ├── experiment.py ├── model.py ├── optimizer.py ├── stu_utils.py ├── synthetic.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/README.md -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/example.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/setup.py -------------------------------------------------------------------------------- /spectral_ssm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/__init__.py -------------------------------------------------------------------------------- /spectral_ssm/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/cifar10.py -------------------------------------------------------------------------------- /spectral_ssm/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/experiment.py -------------------------------------------------------------------------------- /spectral_ssm/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/model.py -------------------------------------------------------------------------------- /spectral_ssm/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/optimizer.py -------------------------------------------------------------------------------- /spectral_ssm/stu_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/stu_utils.py -------------------------------------------------------------------------------- /spectral_ssm/synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/synthetic.py -------------------------------------------------------------------------------- /spectral_ssm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/spectral_ssm/HEAD/spectral_ssm/utils.py --------------------------------------------------------------------------------