All modules for which code is available
- core.data.recording_ids
- core.data.unit_filtering
- core.dataset
- core.features.isi
- core.finetuning.base
- core.finetuning.full_finetuning
- core.finetuning.gradual_unfreezing
- core.finetuning.probe
- core.model
- core.nn.embedding
- core.nn.init
- core.nn.loss
- core.nn.metrics.rankme
- core.nn.multitask_readout
- core.nn.varlen_attention
- core.samplers.distributed_sampler_wrapper
- core.samplers.session_batch_sampler
- core.trainer
- core.transforms.augmentations
- core.transforms.unit_mask
- core.transforms.unit_transforms
- ibl_bwb_eval.entity_ids
- ibl_bwb_eval.metrics
- ibl_bwb_eval.multi_unit
- ibl_bwb_eval.predictions
- ibl_bwb_eval.scoring.aggregation
- ibl_bwb_eval.scoring.ts1
- ibl_bwb_eval.scoring.ts2
- ibl_bwb_eval.scoring.ts3
- ibl_bwb_eval.tasks
- pretrain.datasets.mask_modeling_spike
- pretrain.datasets.multi_task_behavior
- pretrain.datasets.single_task_behavior
- pretrain.models.mtm.masker
- pretrain.models.mtm.mtm
- pretrain.models.mtm.mtm_pretrain
- pretrain.models.ndt2.masker
- pretrain.models.ndt2.ndt2
- pretrain.models.ndt2.ndt2_pretrain
- pretrain.models.ndt_stitch.masker
- pretrain.models.ndt_stitch.ndt_stitch
- pretrain.models.ndt_stitch.ndt_stitch_pretrain
- pretrain.models.neds.masker
- pretrain.models.neds.neds
- pretrain.models.neds.neds_pretrain
- pretrain.models.nemo.nemo
- pretrain.models.nemo.nemo_pretrain
- pretrain.models.nuclr.loss
- pretrain.models.nuclr.nuclr
- pretrain.models.nuclr.nuclr_pretrain
- pretrain.models.possm.possm
- pretrain.models.possm.possm_multitask_pretrain
- pretrain.models.possm.possm_single_task_pretrain
- pretrain.models.poyo.poyo
- pretrain.models.poyo.poyo_single_task_pretrain
- pretrain.models.poyo_plus.poyo_plus
- pretrain.models.poyo_plus.poyo_plus_multitask_pretrain
- pretrain.models.rrr.rrr
- pretrain.models.rrr.rrr_single_task_pretrain
- ts1.models.pretrained.mtm.mtm_eval_trainer
- ts1.models.pretrained.ndt2.ndt2_eval_trainer
- ts1.models.pretrained.ndt_stitch.ndt_stitch_eval_trainer
- ts1.models.pretrained.neds.neds_eval_trainer
- ts1.models.pretrained.possm.possm_eval_trainer
- ts1.models.pretrained.poyo.poyo_eval_trainer
- ts1.models.pretrained.poyo_plus.poyo_plus_eval_trainer
- ts1.models.pretrained.rrr.rrr_eval_trainer
- ts1.models.single_session.cebra.cebra
- ts1.models.single_session.cebra.cebra_eval_trainer
- ts1.models.single_session.gru.gru
- ts1.models.single_session.linear.linear
- ts1.models.single_session.mlp.mlp
- ts1.models.single_session.ndt_superv.ndt_superv
- ts1.models.single_session.tcn.tcn
- ts1.ts1_dataset
- ts1.ts1_eval_trainer
- ts1.ts1_test_mixin
- ts2.models.pretrained.mtm.mtm_eval_trainer
- ts2.models.pretrained.ndt_stitch.ndt_stitch_eval_trainer
- ts2.models.single_session.autoencoder.autoencoder
- ts2.models.single_session.lfads.lfads
- ts2.models.single_session.lfads.lfads_eval_trainer
- ts2.models.single_session.ndt.ndt
- ts2.models.single_session.ndt.ndt_eval_trainer
- ts2.models.single_session.stat_baseline.base
- ts2.models.single_session.stat_baseline.co_smoothing.pop_coupling
- ts2.models.single_session.stat_baseline.co_smoothing.readout_isi
- ts2.models.single_session.stat_baseline.co_smoothing.readout_rrr
- ts2.models.single_session.stat_baseline.forecasting.ridge_ar
- ts2.models.single_session.stat_baseline.forecasting.shrinkage
- ts2.models.single_session.stat_baseline.forecasting.trailing_mean
- ts2.models.single_session.stat_baseline.stat_baseline_trainer
- ts2.ts2_dataset
- ts2.ts2_eval_trainer
- ts2.ts2_test_mixin
- ts3.models.base
- ts3.models.inductive.isi.isi_extractor
- ts3.models.inductive.nemo.nemo_extractor
- ts3.models.inductive.nuclr.nuclr_extractor
- ts3.models.supervised.lolcat.lolcat
- ts3.models.supervised.lolcat.lolcat_trainer
- ts3.models.supervised.lolcat.sampler
- ts3.models.transductive.base
- ts3.models.transductive.mtm.mtm_calibrate_trainer
- ts3.models.transductive.mtm.mtm_extractor
- ts3.models.transductive.ndt_stitch.ndt_stitch_calibrate_trainer
- ts3.models.transductive.ndt_stitch.ndt_stitch_extractor
- ts3.models.transductive.possm.possm_calibrate_trainer
- ts3.models.transductive.possm.possm_extractor
- ts3.models.transductive.poyo_plus.poyo_plus_calibrate_trainer
- ts3.models.transductive.poyo_plus.poyo_plus_extractor
- ts3.probes.base
- ts3.probes.linear
- ts3.probes.mlp
- ts3.ts3_dataset