.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/README.md
docs/make.bat
docs/requirements.txt
docs/source/conf.py
docs/source/index.rst
docs/source/_data/perich_miller_population_2018/c_20131003_center_out_reaching.h5
docs/source/_data/perich_miller_population_2018/c_20131022_center_out_reaching.h5
docs/source/_data/perich_miller_population_2018/c_20131023_center_out_reaching.h5
docs/source/_static/torch_brain_logo.png
docs/source/_static/img/triangle_distribution.png
docs/source/concepts/getting_started.rst
docs/source/concepts/installation.rst
docs/source/concepts/sampler.rst
docs/source/concepts/examples/_utils.py
docs/source/concepts/examples/plot_sampler_1.py
docs/source/concepts/examples/plot_sampler_2.py
docs/source/concepts/examples/plot_sampler_3.py
docs/source/concepts/examples/plot_sampler_4.py
docs/source/concepts/examples/plot_sampler_5.py
docs/source/package/models.rst
docs/source/package/optim.rst
docs/source/package/registry.rst
docs/source/package/transforms.rst
docs/source/package/utils.rst
docs/source/package/data/collate.rst
docs/source/package/data/dataset.rst
docs/source/package/data/index.rst
docs/source/package/data/sampler.rst
docs/source/package/dataset/dataset.rst
docs/source/package/dataset/index.rst
docs/source/package/dataset/mixins.rst
docs/source/package/dataset/nested.rst
docs/source/package/nn/embeddings.rst
docs/source/package/nn/index.rst
docs/source/package/nn/loss.rst
docs/source/package/nn/readout.rst
docs/source/package/nn/transformer.rst
docs/source/tutorials/README.rst
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/train.py
examples/poyo_plus/configs/defaults.yaml
examples/poyo_plus/configs/finetune.yaml
examples/poyo_plus/configs/train_capoyo.yaml
examples/poyo_plus/configs/train_capoyo_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/capoyo.yaml
examples/poyo_plus/configs/dataset/capoyo_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/capoyo.yaml
examples/poyo_plus/configs/model/capoyo_single_session.yaml
examples/poyo_plus/configs/model/poyo_1.yaml
examples/poyo_plus/configs/model/poyo_single_session.yaml
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/capoyo.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/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/gradient_rescale.py
torch_brain/utils/misc.py
torch_brain/utils/readout.py
torch_brain/utils/seed_everything.py
torch_brain/utils/sinusoidal_encoding.py
torch_brain/utils/stitcher.py
torch_brain/utils/tokenizers.py
torch_brain/utils/weights.py