├── .gitignore ├── .pre-commit-config.yaml ├── CITATION.cff ├── LICENSE ├── README.md ├── configs ├── callbacks │ ├── default.yaml │ └── nlb.yaml ├── datamodule │ ├── nlb_area2_bump.yaml │ ├── nlb_dmfc_rsg.yaml │ ├── nlb_mc_maze.yaml │ ├── nlb_mc_maze_large.yaml │ ├── nlb_mc_maze_medium.yaml │ ├── nlb_mc_maze_small.yaml │ ├── nlb_mc_rtt.yaml │ ├── nlb_multisession_mc_maze.yaml │ └── rouse_multisession.yaml ├── model │ ├── nlb_area2_bump.yaml │ ├── nlb_dmfc_rsg.yaml │ ├── nlb_mc_maze.yaml │ ├── nlb_mc_maze_large.yaml │ ├── nlb_mc_maze_medium.yaml │ ├── nlb_mc_maze_small.yaml │ ├── nlb_mc_rtt.yaml │ ├── nlb_multisession_mc_maze.yaml │ ├── rouse_multisession_PCR.yaml │ └── rouse_multisession_random.yaml ├── multi.yaml ├── pbt.yaml └── single.yaml ├── datasets ├── area2_bump-05ms-val.h5 ├── dmfc_rsg-05ms-val.h5 ├── mc_maze-20ms-val.h5 ├── mc_maze_large-05ms-val.h5 ├── mc_maze_medium-05ms-val.h5 ├── mc_maze_small-05ms-val.h5 └── mc_rtt-05ms-val.h5 ├── lfads_torch ├── __init__.py ├── callbacks.py ├── datamodules.py ├── extensions │ ├── __init__.py │ ├── nlb.py │ └── tune.py ├── metrics.py ├── model.py ├── modules │ ├── __init__.py │ ├── augmentations.py │ ├── decoder.py │ ├── encoder.py │ ├── initializers.py │ ├── l2.py │ ├── priors.py │ ├── readin_readout.py │ ├── recons.py │ └── recurrent.py ├── post_run │ ├── __init__.py │ ├── analysis.py │ └── pbt.py ├── run_model.py ├── tuples.py └── utils.py ├── requirements.txt ├── scripts ├── run_multi.py ├── run_pbt.py └── run_single.py ├── setup.py └── tutorials └── multisession ├── 1_data_prep.ipynb ├── 2_run_pbt.py └── 3_evaluation.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/README.md -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/nlb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/callbacks/nlb.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_area2_bump.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_area2_bump.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_dmfc_rsg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_dmfc_rsg.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_mc_maze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_mc_maze.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_mc_maze_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_mc_maze_large.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_mc_maze_medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_mc_maze_medium.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_mc_maze_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_mc_maze_small.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_mc_rtt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_mc_rtt.yaml -------------------------------------------------------------------------------- /configs/datamodule/nlb_multisession_mc_maze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/nlb_multisession_mc_maze.yaml -------------------------------------------------------------------------------- /configs/datamodule/rouse_multisession.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/datamodule/rouse_multisession.yaml -------------------------------------------------------------------------------- /configs/model/nlb_area2_bump.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_area2_bump.yaml -------------------------------------------------------------------------------- /configs/model/nlb_dmfc_rsg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_dmfc_rsg.yaml -------------------------------------------------------------------------------- /configs/model/nlb_mc_maze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_mc_maze.yaml -------------------------------------------------------------------------------- /configs/model/nlb_mc_maze_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_mc_maze_large.yaml -------------------------------------------------------------------------------- /configs/model/nlb_mc_maze_medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_mc_maze_medium.yaml -------------------------------------------------------------------------------- /configs/model/nlb_mc_maze_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_mc_maze_small.yaml -------------------------------------------------------------------------------- /configs/model/nlb_mc_rtt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_mc_rtt.yaml -------------------------------------------------------------------------------- /configs/model/nlb_multisession_mc_maze.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/nlb_multisession_mc_maze.yaml -------------------------------------------------------------------------------- /configs/model/rouse_multisession_PCR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/rouse_multisession_PCR.yaml -------------------------------------------------------------------------------- /configs/model/rouse_multisession_random.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/model/rouse_multisession_random.yaml -------------------------------------------------------------------------------- /configs/multi.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/multi.yaml -------------------------------------------------------------------------------- /configs/pbt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/pbt.yaml -------------------------------------------------------------------------------- /configs/single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/configs/single.yaml -------------------------------------------------------------------------------- /datasets/area2_bump-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/area2_bump-05ms-val.h5 -------------------------------------------------------------------------------- /datasets/dmfc_rsg-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/dmfc_rsg-05ms-val.h5 -------------------------------------------------------------------------------- /datasets/mc_maze-20ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/mc_maze-20ms-val.h5 -------------------------------------------------------------------------------- /datasets/mc_maze_large-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/mc_maze_large-05ms-val.h5 -------------------------------------------------------------------------------- /datasets/mc_maze_medium-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/mc_maze_medium-05ms-val.h5 -------------------------------------------------------------------------------- /datasets/mc_maze_small-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/mc_maze_small-05ms-val.h5 -------------------------------------------------------------------------------- /datasets/mc_rtt-05ms-val.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/datasets/mc_rtt-05ms-val.h5 -------------------------------------------------------------------------------- /lfads_torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lfads_torch/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/callbacks.py -------------------------------------------------------------------------------- /lfads_torch/datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/datamodules.py -------------------------------------------------------------------------------- /lfads_torch/extensions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lfads_torch/extensions/nlb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/extensions/nlb.py -------------------------------------------------------------------------------- /lfads_torch/extensions/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/extensions/tune.py -------------------------------------------------------------------------------- /lfads_torch/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/metrics.py -------------------------------------------------------------------------------- /lfads_torch/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/model.py -------------------------------------------------------------------------------- /lfads_torch/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lfads_torch/modules/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/augmentations.py -------------------------------------------------------------------------------- /lfads_torch/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/decoder.py -------------------------------------------------------------------------------- /lfads_torch/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/encoder.py -------------------------------------------------------------------------------- /lfads_torch/modules/initializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/initializers.py -------------------------------------------------------------------------------- /lfads_torch/modules/l2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/l2.py -------------------------------------------------------------------------------- /lfads_torch/modules/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/priors.py -------------------------------------------------------------------------------- /lfads_torch/modules/readin_readout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/readin_readout.py -------------------------------------------------------------------------------- /lfads_torch/modules/recons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/recons.py -------------------------------------------------------------------------------- /lfads_torch/modules/recurrent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/modules/recurrent.py -------------------------------------------------------------------------------- /lfads_torch/post_run/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lfads_torch/post_run/analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/post_run/analysis.py -------------------------------------------------------------------------------- /lfads_torch/post_run/pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/post_run/pbt.py -------------------------------------------------------------------------------- /lfads_torch/run_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/run_model.py -------------------------------------------------------------------------------- /lfads_torch/tuples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/tuples.py -------------------------------------------------------------------------------- /lfads_torch/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/lfads_torch/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/run_multi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/scripts/run_multi.py -------------------------------------------------------------------------------- /scripts/run_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/scripts/run_pbt.py -------------------------------------------------------------------------------- /scripts/run_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/scripts/run_single.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/setup.py -------------------------------------------------------------------------------- /tutorials/multisession/1_data_prep.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/tutorials/multisession/1_data_prep.ipynb -------------------------------------------------------------------------------- /tutorials/multisession/2_run_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/tutorials/multisession/2_run_pbt.py -------------------------------------------------------------------------------- /tutorials/multisession/3_evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arsedler9/lfads-torch/HEAD/tutorials/multisession/3_evaluation.ipynb --------------------------------------------------------------------------------