.gitattributes
.gitignore
.pre-commit-config.yaml
.readthedocs.yaml
CHANGELOG.md
CONTRIBUTING.md
GOVERNANCE.md
LICENSE.txt
README.md
RELEASE.md
environment.yaml
pyproject.toml
.github/workflows/benchmarks.yml
.github/workflows/linting.yml
.github/workflows/pipeline_testing.yml
.github/workflows/pipeline_testing_matrix.yml
.github/workflows/publish.yml
.github/workflows/testing.yml
.github/workflows/typecheck.yml
docs/Makefile
docs/make.bat
docs/source/api_reference.py
docs/source/conf.py
docs/source/index.rst
docs/source/installation.rst
docs/source/_static/gappy-slice.png
docs/source/_static/temporaldata_logo.png
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/_templates/data_objects.rst
docs/source/_templates/dataset.rst
docs/source/api/index.rst.template
docs/source/api/module.rst.template
docs/source/cli/commands.rst
docs/source/guides/index.rst
docs/source/guides/data/data_objects_intro.rst
docs/source/guides/data/domain.rst
docs/source/guides/data/gappy_regular_ts.rst
docs/source/guides/data/index.rst
docs/source/guides/data/io.rst
docs/source/guides/data/interval_ops/index.rst
docs/source/guides/data/interval_ops/plots.py
docs/source/guides/examples/index.rst
docs/source/guides/examples/neuroprobe.rst
docs/source/guides/pipeline/create_pipeline.rst
docs/source/guides/pipeline/index.rst
docs/source/guides/pipeline/openneuro_pipeline.rst
docs/source/guides/pipeline/prepare_data.rst
docs/source/guides/pipeline/using_existing_data.rst
docs/source/guides/training/index.rst
docs/source/guides/training/sampling/_utils.py
docs/source/guides/training/sampling/index.rst
docs/source/guides/training/sampling/static/overview.png
docs/source/guides/training/sampling/static/sampler_10_0.png
docs/source/guides/training/sampling/static/sampler_10s.gif
docs/source/guides/training/sampling/static/sampler_12_0.png
docs/source/guides/training/sampling/static/sampler_1s.gif
docs/source/guides/training/sampling/static/sampler_multi.gif
docs/source/guides/training/sampling/static/sampler_sequential_10s.gif
docs/source/guides/training/sampling/static/sampler_trial_10s.gif
docs/source/js/scripts/api-search.js
docs/source/notebooks/nlb_maze_minimal_example/modify.py
docs/source/notebooks/nlb_maze_minimal_example/nlb_maze_minimal_example.ipynb
docs/source/scss/api-search.scss
docs/source/scss/api.scss
docs/source/sphinxext/autoshortsummary.py
examples/nlb_maze_minimal/README.md
examples/nlb_maze_minimal/models.py
examples/nlb_maze_minimal/train.py
examples/poyo/README.md
examples/poyo/optim.py
examples/poyo/train_simple.py
examples/poyo/utils.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/__init__.py
examples/poyo/datasets/nlb.py
examples/poyo/datasets/poyo_1.py
examples/poyo/datasets/poyo_mp.py
examples/poyo/datasets/wrapper.py
howto/MIGRATE_TO_v0.2.md
scripts/data_benchmarks/benchmark.py
scripts/data_benchmarks/compare.py
tests/_test_old_dataset_sim.py
tests/test_binning.py
tests/test_collate.py
tests/test_infinite_vocab_embedding.py
tests/test_new_dataset.py
tests/test_poyo.py
tests/test_random_time_scaling.py
tests/test_rotary_attention.py
tests/test_sampler.py
tests/test_stitch.py
tests/test_stitcher_sampler.py
tests/test_unit_dropout.py
tests/test_unit_filter.py
tests/test_version.py
tests/cli/test_cli.py
tests/data/test_arraydict.py
tests/data/test_concat.py
tests/data/test_data.py
tests/data/test_descriptions.py
tests/data/test_file_handle.py
tests/data/test_h5_io.py
tests/data/test_interval.py
tests/data/test_irregular_ts.py
tests/data/test_materialize.py
tests/data/test_regular_ts.py
tests/data/test_select_by_mask.py
tests/data/test_serialize.py
tests/datasets/test_neuroprobe2025_dataset.py
tests/datasets/test_openneuro_dataset.py
tests/pipeline/_utils.py
tests/pipeline/test_neuroprobe2025_pipeline.py
tests/pipeline/test_openneuro_pipeline.py
tests/pipeline/test_read_inline_metadata.py
tests/pipeline/test_vollan_pipeline.py
tests/utils/test_bids.py
tests/utils/test_misc.py
tests/utils/test_mne.py
tests/utils/test_openneuro.py
tests/utils/test_s3.py
tests/utils/test_signal.py
tests/utils/test_split.py
torch_brain/__init__.py
torch_brain.egg-info/PKG-INFO
torch_brain.egg-info/SOURCES.txt
torch_brain.egg-info/dependency_links.txt
torch_brain.egg-info/entry_points.txt
torch_brain.egg-info/requires.txt
torch_brain.egg-info/top_level.txt
torch_brain/batching/__init__.py
torch_brain/batching/collate.py
torch_brain/data/__init__.py
torch_brain/data/arraydict.py
torch_brain/data/collate.py
torch_brain/data/concat.py
torch_brain/data/data.py
torch_brain/data/dataset.py
torch_brain/data/descriptions.py
torch_brain/data/interval.py
torch_brain/data/irregular_ts.py
torch_brain/data/regular_ts.py
torch_brain/data/sampler.py
torch_brain/data/serialization.py
torch_brain/data/typing.py
torch_brain/data/utils.py
torch_brain/dataset/__init__.py
torch_brain/datasets/AllenVisualCodingOphys2016.py
torch_brain/datasets/ChurchlandShenoyNeural2012.py
torch_brain/datasets/FlintSlutzkyAccurate2012.py
torch_brain/datasets/KempSleepEDF2013.py
torch_brain/datasets/KlinzingSleepDS005555.py
torch_brain/datasets/KochiVisualNamingDS006914.py
torch_brain/datasets/Neuroprobe2025.py
torch_brain/datasets/OdohertySabesNonhuman2017.py
torch_brain/datasets/OpenNeuroDataset.py
torch_brain/datasets/PeiPandarinathNLB2021.py
torch_brain/datasets/PerichMillerPopulation2018.py
torch_brain/datasets/ShiraziHBNR1DS005505.py
torch_brain/datasets/VollanMoserAlternating2025.py
torch_brain/datasets/__init__.py
torch_brain/datasets/_utils.py
torch_brain/datasets/dataset.py
torch_brain/datasets/mixins.py
torch_brain/datasets/nested.py
torch_brain/models/__init__.py
torch_brain/models/poyo.py
torch_brain/nn/__init__.py
torch_brain/nn/infinite_vocab_embedding.py
torch_brain/nn/position_embeddings.py
torch_brain/nn/rotary_attention.py
torch_brain/pipeline/__init__.py
torch_brain/pipeline/config.py
torch_brain/pipeline/openneuro.py
torch_brain/pipeline/pipeline.py
torch_brain/pipeline/runner.py
torch_brain/pipeline/_cli/__init__.py
torch_brain/pipeline/_cli/cli.py
torch_brain/pipeline/_cli/cli_completion.py
torch_brain/pipeline/_cli/cli_config.py
torch_brain/pipeline/_cli/cli_list.py
torch_brain/pipeline/_cli/cli_prepare.py
torch_brain/pipeline/_cli/utils.py
torch_brain/pipeline/brainsets-pipelines/allen_visual_coding_ophys_2016/pipeline.py
torch_brain/pipeline/brainsets-pipelines/allen_visual_coding_ophys_2016/session_ids.txt
torch_brain/pipeline/brainsets-pipelines/allen_visual_coding_ophys_2016/split.py
torch_brain/pipeline/brainsets-pipelines/churchland_shenoy_neural_2012/pipeline.py
torch_brain/pipeline/brainsets-pipelines/flint_slutzky_accurate_2012/pipeline.py
torch_brain/pipeline/brainsets-pipelines/kemp_sleep_edf_2013/pipeline.py
torch_brain/pipeline/brainsets-pipelines/klinzing_sleep_ds005555/pipeline.py
torch_brain/pipeline/brainsets-pipelines/kochi_visualnaming_ds006914/pipeline.py
torch_brain/pipeline/brainsets-pipelines/neuroprobe_2025/pipeline.py
torch_brain/pipeline/brainsets-pipelines/odoherty_sabes_nonhuman_2017/pipeline.py
torch_brain/pipeline/brainsets-pipelines/pei_pandarinath_nlb_2021/README.md
torch_brain/pipeline/brainsets-pipelines/pei_pandarinath_nlb_2021/pipeline.py
torch_brain/pipeline/brainsets-pipelines/perich_miller_population_2018/pipeline.py
torch_brain/pipeline/brainsets-pipelines/shirazi_hbnr1_ds005505/pipeline.py
torch_brain/pipeline/brainsets-pipelines/vollan_moser_alternating_2025/pipeline.py
torch_brain/samplers/__init__.py
torch_brain/samplers/distributed_evaluation_sampler.py
torch_brain/samplers/distributed_stitching_fixed_window.py
torch_brain/samplers/random_fixed_window.py
torch_brain/samplers/sequential_fixed_window.py
torch_brain/samplers/trial_sampler.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/bids.py
torch_brain/utils/binning.py
torch_brain/utils/callbacks.py
torch_brain/utils/dandi.py
torch_brain/utils/misc.py
torch_brain/utils/mne.py
torch_brain/utils/openneuro.py
torch_brain/utils/s3.py
torch_brain/utils/signal.py
torch_brain/utils/split.py
torch_brain/utils/stitcher.py
torch_brain/utils/weights.py