.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CHANGELOG.md
CONTRIBUTING.md
LICENSE.txt
MANIFEST.in
README.md
RELEASE.md
environment.yaml
pyproject.toml
.github/workflows/linting.yml
.github/workflows/publish.yml
.github/workflows/testing.yml
docs/Makefile
docs/make.bat
docs/source/api_reference.py
docs/source/conf.py
docs/source/index.rst
docs/source/_static/torch_brain_logo.png
docs/source/_static/css/custom.css
docs/source/_static/img/triangle_distribution.png
docs/source/_templates/base.rst
docs/source/api/index.rst.template
docs/source/api/module.rst.template
docs/source/guides/index.rst
docs/source/guides/installation.rst
docs/source/guides/sampling/_utils.py
docs/source/guides/sampling/index.rst
docs/source/guides/sampling/plot_sampler_1.py
docs/source/guides/sampling/plot_sampler_2.py
docs/source/guides/sampling/plot_sampler_3.py
docs/source/guides/sampling/plot_sampler_4.py
docs/source/guides/sampling/plot_sampler_5.py
docs/source/js/scripts/api-search.js
docs/source/scss/api-search.scss
docs/source/scss/api.scss
docs/source/sphinxext/autoshortsummary.py
examples/poyo/README.md
examples/poyo/train.py
examples/poyo/configs/defaults.yaml
examples/poyo/configs/train_mc_maze_small.yaml
examples/poyo/configs/train_poyo_1.yaml
examples/poyo/configs/train_poyo_mp.yaml
examples/poyo/configs/model/poyo_1.3M.yaml
examples/poyo/configs/model/poyo_11.8M.yaml
examples/poyo/datasets/nlb.py
examples/poyo/datasets/poyo_1.py
examples/poyo/datasets/poyo_mp.py
examples/poyo_plus/README.md
examples/poyo_plus/finetune.py
examples/poyo_plus/metrics.py
examples/poyo_plus/train.py
examples/poyo_plus/configs/defaults.yaml
examples/poyo_plus/configs/finetune.yaml
examples/poyo_plus/configs/train_calcium_poyo_plus.yaml
examples/poyo_plus/configs/train_calcium_poyo_plus_single_session.yaml
examples/poyo_plus/configs/train_mc_maze_small.yaml
examples/poyo_plus/configs/train_poyo_mp.yaml
examples/poyo_plus/configs/dataset/calcium_poyo_plus_example.yaml
examples/poyo_plus/configs/dataset/calcium_poyo_plus_single_session.yaml
examples/poyo_plus/configs/dataset/pei_pandarinath_nlb_2021.yaml
examples/poyo_plus/configs/dataset/perich_miller_population_2018.yaml
examples/poyo_plus/configs/model/calcium_poyo_plus.yaml
examples/poyo_plus/configs/model/calcium_poyo_plus_single_session.yaml
examples/poyo_plus/configs/model/poyo_1.yaml
examples/poyo_plus/configs/model/poyo_single_session.yaml
examples/poyo_plus/scripts/generate_config.py
examples/poyo_plus/scripts/task_contents.csv
notebooks/model_writing_tutorial.ipynb
scripts/calculate_normalization_scales.py
tests/test_binning.py
tests/test_collate.py
tests/test_dataset_real.py
tests/test_dataset_sim.py
tests/test_infinite_vocab_embedding.py
tests/test_loss.py
tests/test_multitask_decoding_stitch_evaluator.py
tests/test_new_dataset.py
tests/test_poyo.py
tests/test_poyo_plus.py
tests/test_random_time_scaling.py
tests/test_registry.py
tests/test_rotary_attention.py
tests/test_sampler.py
tests/test_stitch.py
tests/test_stitcher_sampler.py
tests/test_tokenizers.py
tests/test_unit_dropout.py
tests/test_unit_filter.py
tests/test_validation.py
tests/test_version.py
tests/test_weights.py
tests/test_zscaling_script.py
torch_brain/__init__.py
torch_brain/optim.py
torch_brain/registry.py
torch_brain.egg-info/PKG-INFO
torch_brain.egg-info/SOURCES.txt
torch_brain.egg-info/dependency_links.txt
torch_brain.egg-info/requires.txt
torch_brain.egg-info/top_level.txt
torch_brain/data/__init__.py
torch_brain/data/collate.py
torch_brain/data/dataset.py
torch_brain/data/sampler.py
torch_brain/dataset/__init__.py
torch_brain/dataset/dataset.py
torch_brain/dataset/mixins.py
torch_brain/dataset/nested.py
torch_brain/models/__init__.py
torch_brain/models/calcium_poyo_plus.py
torch_brain/models/poyo.py
torch_brain/models/poyo_plus.py
torch_brain/nn/__init__.py
torch_brain/nn/embedding.py
torch_brain/nn/feedforward.py
torch_brain/nn/infinite_vocab_embedding.py
torch_brain/nn/loss.py
torch_brain/nn/multitask_readout.py
torch_brain/nn/position_embeddings.py
torch_brain/nn/rotary_attention.py
torch_brain/transforms/__init__.py
torch_brain/transforms/bin_spikes.py
torch_brain/transforms/container.py
torch_brain/transforms/output_sampler.py
torch_brain/transforms/random_crop.py
torch_brain/transforms/random_time_scaling.py
torch_brain/transforms/unit_dropout.py
torch_brain/transforms/unit_filter.py
torch_brain/utils/__init__.py
torch_brain/utils/binning.py
torch_brain/utils/callbacks.py
torch_brain/utils/misc.py
torch_brain/utils/readout.py
torch_brain/utils/seed_everything.py
torch_brain/utils/stitcher.py
torch_brain/utils/tokenizers.py
torch_brain/utils/weights.py