Index A | B | C | D | E | F | G | I | K | L | M | N | P | R | S | T | U | V | W A ACGEncoder (class in pretrain.models) add_checkpoint_items() (core.trainer.BaseTrainer method) AdditiveGaussianNoise (class in core.transforms) AdditivePepperNoise (class in core.transforms) aggregate() (in module ibl_bwb_eval.scoring.aggregation) aggregate_metrics() (in module ibl_bwb_eval.metrics) AmplitudeScaling (class in core.transforms) apply_unit_mask_() (in module core.transforms) AutoencoderMLP (class in ts2.models.single_session) B barrier() (core.trainer.BaseTrainer method) BaseModel (class in core.model) BaseTrainer (class in core.trainer) BINARY (ibl_bwb_eval.DataType attribute) BPS (class in ibl_bwb_eval.metrics) bs (pretrain.models.RRRDecoder attribute) BuildUnitFiltering (class in core.data) C CEBRA (class in ts1.models.single_session) CEBRAEvalTrainer (class in ts1.models.single_session) check_ts3_label_order() (in module ibl_bwb_eval) clip_and_log_grad_norm() (core.trainer.BaseTrainer method) CLIPLoss (class in core.nn.loss) compute() (ibl_bwb_eval.metrics.BPS method) (ibl_bwb_eval.metrics.PoissonD2Score method) compute_bin_index() (pretrain.models.POSSM method) compute_isi_histogram() (in module ts3) compute_multitask_loss() (pretrain.models.POSSMMultitaskPretrain method) configure_multitask_readout() (pretrain.models.POSSM method) configure_readout() (core.model.BaseModel method) (pretrain.models.MtM method) (pretrain.models.NDT2 method) (pretrain.models.NDTStitch method) (pretrain.models.POSSM method) (pretrain.models.POYO method) (pretrain.models.POYOPlus method) (pretrain.models.RRRDecoder method) (ts1.models.single_session.CEBRA method) (ts1.models.single_session.GRU method) (ts1.models.single_session.Linear method) (ts1.models.single_session.MLP method) (ts1.models.single_session.NDTSuperv method) (ts1.models.single_session.TCN method) CONTINUOUS (ibl_bwb_eval.DataType attribute) core.data module core.dataset module core.finetuning module core.model module core.nn module core.nn.loss module core.nn.metrics module core.samplers module core.trainer module core.transforms module create_search_space() (core.model.BaseModel class method) (pretrain.models.POSSM class method) (pretrain.models.POYO class method) (pretrain.models.RRRDecoder class method) (ts1.models.single_session.CEBRA class method) (ts1.models.single_session.GRU class method) (ts1.models.single_session.Linear class method) (ts1.models.single_session.MLP class method) (ts1.models.single_session.NDTSuperv class method) (ts1.models.single_session.TCN class method) (ts2.models.single_session.AutoencoderMLP class method) (ts2.models.single_session.LFADS class method) (ts2.models.single_session.NDT class method) cross_attn() (in module core.nn) custom_init() (pretrain.models.MtM method) (pretrain.models.NDTStitch method) (ts1.models.single_session.NDTSuperv method) (ts2.models.single_session.NDT method) D data_type (ibl_bwb_eval.TS1ReadoutSpec attribute) dataset_transform() (core.dataset.IBLBrainWideBench2026 method) (core.dataset.WholeSessionSpikeDataset method) (pretrain.datasets.IBLBrainWideBenchMultiTaskBehavior method) (pretrain.datasets.IBLBrainWideBenchSingleTaskBehavior method) (ts1.IBLBrainWideBenchTS1 method) (ts2.IBLBrainWideBenchTS2 method) dataset_version (core.dataset.IBLBrainWideBench2026 property) DataType (class in ibl_bwb_eval) decode_entity_ids() (in module ibl_bwb_eval.entity_ids) dim (ibl_bwb_eval.TS1ReadoutSpec attribute) (ibl_bwb_eval.TS3ReadoutSpec property) DistributedSamplerWrapper (class in core.samplers) domain_key (ibl_bwb_eval.TS1ReadoutSpec attribute) E Embedding (class in core.nn) encode() (ts3.Extractor method) (ts3.models.inductive.ISIExtractor method) (ts3.models.inductive.NEMOExtractor method) (ts3.models.inductive.NuCLRExtractor method) (ts3.models.transductive.POSSMExtractor method) (ts3.models.transductive.POYOPlusExtractor method) (ts3.models.transductive.TransductiveExtractor method) encode_entity_ids() (in module ibl_bwb_eval.entity_ids) enforce_unit_filtering() (in module core.data) EVENT_RATE (ibl_bwb_eval.DataType attribute) Extractor (class in ts3) F FilterUnits (class in core.transforms) FinetuningStrategy (class in core.finetuning) fit() (ts1.models.single_session.CEBRA method) (ts2.models.single_session.PopCoupling method) (ts2.models.single_session.RidgeAR method) (ts2.models.single_session.RRRReadout method) (ts2.models.single_session.RRRReadoutWithISI method) (ts2.models.single_session.Shrinkage method) (ts2.models.single_session.StatBaseline method) (ts2.models.single_session.TrailingMean method) fit_predict() (ts3.probes.LinearProbe method) (ts3.probes.MLPProbe method) (ts3.probes.Probe method) forward() (core.model.BaseModel method) (core.nn.loss.CLIPLoss method) (core.nn.loss.WeightedMSELoss method) (core.nn.MultitaskReadout method) (pretrain.models.ACGEncoder method) (pretrain.models.LinearProjector method) (pretrain.models.MtM method) (pretrain.models.MtMMasker method) (pretrain.models.NDT2 method) (pretrain.models.NDT2Masker method) (pretrain.models.NDTStitch method) (pretrain.models.NDTStitchMasker method) (pretrain.models.NEDS method) (pretrain.models.NEDSMasker method) (pretrain.models.NEMO method) (pretrain.models.NuCLR method) (pretrain.models.NuCLRLoss method) (pretrain.models.POSSM method) (pretrain.models.POYO method) (pretrain.models.POYOPlus method) (pretrain.models.RRRDecoder method) (pretrain.models.WVFEncoder method) (ts1.models.single_session.CEBRA method) (ts1.models.single_session.GRU method) (ts1.models.single_session.Linear method) (ts1.models.single_session.MLP method) (ts1.models.single_session.NDTSuperv method) (ts1.models.single_session.TCN method) (ts2.models.single_session.AutoencoderMLP method) (ts2.models.single_session.LFADS method) (ts2.models.single_session.NDT method) (ts2.models.single_session.StatBaseline method) (ts3.models.supervised.LOLCAT method) (ts3.models.supervised.MultiHeadGlobalAttention method) forward_varlen() (core.nn.MultitaskReadout method) from_ts3() (in module ibl_bwb_eval.scoring.aggregation) FullFinetuning (class in core.finetuning) G GaussianNoise (class in core.transforms) generate_block_mask() (pretrain.models.NDTStitchMasker method) get_best_pbar_metrics() (core.trainer.BaseTrainer method) (pretrain.models.MtMPretrain method) (pretrain.models.NDT2Pretrain method) (pretrain.models.NDTStitchPretrain method) (pretrain.models.POSSMMultitaskPretrain method) get_brain_regions() (core.dataset.IBLBrainWideBench2026 method) get_param_groups() (core.trainer.BaseTrainer method) get_sampling_intervals() (core.dataset.IBLBrainWideBench2026 method) (core.dataset.WholeSessionSpikeDataset method) (ts1.IBLBrainWideBenchTS1 method) (ts2.IBLBrainWideBenchTS2 method) get_trial_intervals() (pretrain.datasets.IBLBrainWideBenchMultiTaskBehavior method) get_ts1_readout_spec() (in module ibl_bwb_eval) get_ts1_supported_tasks() (in module ibl_bwb_eval) (ts1.IBLBrainWideBenchTS1 class method) get_ts2_readout_spec() (in module ibl_bwb_eval) get_ts3_readout_spec() (in module ibl_bwb_eval) GradualUnfreezing (class in core.finetuning) GRU (class in ts1.models.single_session) I ibl_bwb_eval module ibl_bwb_eval.entity_ids module ibl_bwb_eval.metrics module ibl_bwb_eval.multi_unit module ibl_bwb_eval.predictions module ibl_bwb_eval.scoring.aggregation module ibl_bwb_eval.scoring.ts1 module ibl_bwb_eval.scoring.ts2 module ibl_bwb_eval.scoring.ts3 module IBLBrainWideBench2026 (class in core.dataset) IBLBrainWideBenchMaskModelingSpikes (class in pretrain.datasets) IBLBrainWideBenchMultiTaskBehavior (class in pretrain.datasets) IBLBrainWideBenchSingleTaskBehavior (class in pretrain.datasets) IBLBrainWideBenchTS1 (class in ts1) IBLBrainWideBenchTS2 (class in ts2) IBLBrainWideBenchTS3 (class in ts3) id (ibl_bwb_eval.TS1ReadoutSpec attribute) (ibl_bwb_eval.TS2ReadoutSpec attribute) (ibl_bwb_eval.TS3ReadoutSpec attribute) input_fn() (core.model.BaseModel method) (pretrain.models.MtM method) (pretrain.models.NDT2 method) (pretrain.models.NDTStitch method) (pretrain.models.NEDS method) (pretrain.models.POSSM method) (pretrain.models.POYO method) (pretrain.models.POYOPlus method) (pretrain.models.RRRDecoder method) (ts1.models.single_session.CEBRA method) (ts1.models.single_session.GRU method) (ts1.models.single_session.Linear method) (ts1.models.single_session.MLP method) (ts1.models.single_session.NDTSuperv method) (ts1.models.single_session.TCN method) (ts2.models.single_session.AutoencoderMLP method) (ts2.models.single_session.LFADS method) (ts2.models.single_session.NDT method) (ts2.models.single_session.RRRReadoutWithISI method) (ts2.models.single_session.StatBaseline method) interval_key (ibl_bwb_eval.TS1ReadoutSpec attribute) is_task_of() (in module ibl_bwb_eval) ISIExtractor (class in ts3.models.inductive) K keep_mask() (core.dataset.UnitQCPolicy method) L label_names (ibl_bwb_eval.TS3ReadoutSpec attribute) LFADS (class in ts2.models.single_session) LFADSEvalTrainer (class in ts2.models.single_session) Linear (class in ts1.models.single_session) LinearProbe (class in ts3.probes) LinearProjector (class in pretrain.models) link_datasets() (core.model.BaseModel method) (pretrain.models.MtM method) (pretrain.models.NDT2 method) (pretrain.models.NDTStitch method) (pretrain.models.NEDS method) (pretrain.models.POSSM method) (pretrain.models.POYO method) (pretrain.models.POYOPlus method) (pretrain.models.RRRDecoder method) (ts1.models.single_session.NDTSuperv method) (ts2.models.single_session.AutoencoderMLP method) (ts2.models.single_session.LFADS method) (ts2.models.single_session.NDT method) (ts2.models.single_session.StatBaseline method) link_model() (ts1.models.pretrained.MtMEvalTrainer method) (ts1.models.pretrained.NDT2EvalTrainer method) (ts1.models.pretrained.NDTStitchEvalTrainer method) (ts1.models.pretrained.NEDSEvalTrainer method) (ts1.models.pretrained.POSSMEvalTrainer method) (ts1.models.pretrained.POYOEvalTrainer method) (ts1.models.pretrained.POYOPlusEvalTrainer method) (ts1.models.pretrained.RRREvalTrainer method) (ts1.TS1EvalTrainer method) (ts2.models.pretrained.MtMEvalTrainer method) (ts2.models.pretrained.NDTStitchEvalTrainer method) (ts2.models.single_session.StatBaselineTrainer method) (ts2.TS2EvalTrainer method) load_ckpt() (core.model.BaseModel method) (pretrain.models.MtM method) (pretrain.models.NDT2 method) (pretrain.models.NDTStitch method) (pretrain.models.NEDS method) (pretrain.models.POSSM method) (pretrain.models.POYO method) (pretrain.models.POYOPlus method) (pretrain.models.RRRDecoder method) log_epoch_grad_norm() (core.trainer.BaseTrainer method) log_lr() (core.trainer.BaseTrainer method) log_param_grad_stats() (core.trainer.BaseTrainer method) log_training_results() (core.trainer.BaseTrainer method) (pretrain.models.MtMPretrain method) (pretrain.models.NDT2Pretrain method) (pretrain.models.NDTStitchPretrain method) (pretrain.models.NEDSPretrain method) (pretrain.models.POSSMMultitaskPretrain method) (pretrain.models.POYOPlusMultitaskPretrain method) (pretrain.models.RRRSingleTaskPretrain method) (ts1.TS1EvalTrainer method) (ts2.TS2EvalTrainer method) LOLCAT (class in ts3.models.supervised) LOLCATTrainer (class in ts3.models.supervised) loss() (ts1.TS1EvalTrainer method) (ts2.TS2EvalTrainer method) LossFeedbackSampler (class in ts3.models.supervised) M make_ddp() (core.trainer.BaseTrainer method) mask_counts_key (ibl_bwb_eval.TS1ReadoutSpec attribute) mask_dim (ibl_bwb_eval.TS2ReadoutSpec attribute) mask_key (ibl_bwb_eval.TS1ReadoutSpec attribute) metrics (ibl_bwb_eval.TS1ReadoutSpec attribute) (ibl_bwb_eval.TS2ReadoutSpec attribute) (ibl_bwb_eval.TS3ReadoutSpec attribute) MLP (class in ts1.models.single_session) MLPProbe (class in ts3.probes) module core.data core.dataset core.finetuning core.model core.nn core.nn.loss core.nn.metrics core.samplers core.trainer core.transforms ibl_bwb_eval 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 pretrain.datasets pretrain.models ts1 ts1.models.pretrained ts1.models.single_session ts2 ts2.models.pretrained ts2.models.single_session ts3 ts3.models.inductive ts3.models.supervised ts3.models.transductive ts3.probes MtM (class in pretrain.models) MtMCalibrateTrainer (class in ts3.models.transductive) MtMEvalTrainer (class in ts1.models.pretrained) (class in ts2.models.pretrained) MtMExtractor (class in ts3.models.transductive) MtMMasker (class in pretrain.models) MtMPretrain (class in pretrain.models) multi_unit_prediction() (in module ibl_bwb_eval.multi_unit) MultiHeadGlobalAttention (class in ts3.models.supervised) MULTINOMIAL (ibl_bwb_eval.DataType attribute) MultitaskReadout (class in core.nn) N name (ts3.Extractor property) (ts3.models.inductive.ISIExtractor property) (ts3.models.inductive.NEMOExtractor property) (ts3.models.inductive.NuCLRExtractor property) (ts3.models.transductive.POSSMExtractor property) (ts3.models.transductive.POYOPlusExtractor property) (ts3.models.transductive.TransductiveExtractor property) NDT (class in ts2.models.single_session) NDT2 (class in pretrain.models) NDT2EvalTrainer (class in ts1.models.pretrained) NDT2Masker (class in pretrain.models) NDT2Pretrain (class in pretrain.models) NDTEvalTrainer (class in ts2.models.single_session) NDTStitch (class in pretrain.models) NDTStitchCalibrateTrainer (class in ts3.models.transductive) NDTStitchEvalTrainer (class in ts1.models.pretrained) (class in ts2.models.pretrained) NDTStitchExtractor (class in ts3.models.transductive) NDTStitchMasker (class in pretrain.models) NDTStitchPretrain (class in pretrain.models) NDTSuperv (class in ts1.models.single_session) NEDS (class in pretrain.models) NEDSEvalTrainer (class in ts1.models.pretrained) NEDSMasker (class in pretrain.models) NEDSPretrain (class in pretrain.models) NEMO (class in pretrain.models) NEMOExtractor (class in ts3.models.inductive) NEMOPretrain (class in pretrain.models) NuCLR (class in pretrain.models) NuCLRExtractor (class in ts3.models.inductive) NuCLRLoss (class in pretrain.models) NuCLRPretrain (class in pretrain.models) num_timesteps (ibl_bwb_eval.TS1ReadoutSpec property) (ibl_bwb_eval.TS3ReadoutSpec property) P PoissonD2Score (class in ibl_bwb_eval.metrics) PopCoupling (class in ts2.models.single_session) POSSM (class in pretrain.models) POSSMCalibrateTrainer (class in ts3.models.transductive) POSSMEvalTrainer (class in ts1.models.pretrained) POSSMExtractor (class in ts3.models.transductive) POSSMMultitaskPretrain (class in pretrain.models) POSSMSingleTaskPretrain (class in pretrain.models) POYO (class in pretrain.models) POYOEvalTrainer (class in ts1.models.pretrained) POYOPlus (class in pretrain.models) POYOPlusCalibrateTrainer (class in ts3.models.transductive) POYOPlusEvalTrainer (class in ts1.models.pretrained) POYOPlusExtractor (class in ts3.models.transductive) POYOPlusMultitaskPretrain (class in pretrain.models) POYOSingleTaskPretrain (class in pretrain.models) predict() (ts1.models.pretrained.MtMEvalTrainer method) (ts1.models.pretrained.NEDSEvalTrainer method) (ts1.models.pretrained.POSSMEvalTrainer method) (ts1.models.pretrained.POYOEvalTrainer method) (ts1.models.pretrained.POYOPlusEvalTrainer method) (ts1.TS1EvalTrainer method) (ts2.models.pretrained.MtMEvalTrainer method) (ts2.models.single_session.PopCoupling method) (ts2.models.single_session.RidgeAR method) (ts2.models.single_session.RRRReadout method) (ts2.models.single_session.RRRReadoutWithISI method) (ts2.models.single_session.Shrinkage method) (ts2.models.single_session.StatBaseline method) (ts2.models.single_session.TrailingMean method) (ts2.TS2EvalTrainer method) PredictionsWriter (class in ibl_bwb_eval.predictions) pretrain.datasets module pretrain.models module primary_metric (ibl_bwb_eval.TS1ReadoutSpec attribute) (ibl_bwb_eval.TS2ReadoutSpec attribute) Probe (class in core.finetuning) (class in ts3.probes) process_tunable_params() (core.model.BaseModel class method) (pretrain.models.POSSM class method) (pretrain.models.POYO class method) (pretrain.models.RRRDecoder class method) (ts1.models.single_session.CEBRA class method) (ts1.models.single_session.GRU class method) (ts1.models.single_session.Linear class method) (ts1.models.single_session.MLP class method) (ts1.models.single_session.NDTSuperv class method) (ts1.models.single_session.TCN class method) (ts2.models.single_session.AutoencoderMLP class method) (ts2.models.single_session.LFADS class method) (ts2.models.single_session.NDT class method) push_logs() (core.trainer.BaseTrainer method) R RandomApply (class in core.transforms) rank() (in module ibl_bwb_eval.scoring.aggregation) rankme() (in module core.nn.metrics) read() (ts3.models.transductive.MtMExtractor method) (ts3.models.transductive.NDTStitchExtractor method) (ts3.models.transductive.POSSMExtractor method) (ts3.models.transductive.POYOPlusExtractor method) (ts3.models.transductive.TransductiveExtractor method) read_build_unit_filtering() (in module core.data) read_recording_ids() (in module core.data) ReadoutSpec (class in ibl_bwb_eval) reduce_mean() (core.trainer.BaseTrainer method) region_key (ibl_bwb_eval.TS3ReadoutSpec attribute) representation() (pretrain.models.NEMO method) reset_best_tracking() (core.trainer.BaseTrainer method) reset_parameters() (core.nn.Embedding method) RidgeAR (class in ts2.models.single_session) RRRDecoder (class in pretrain.models) RRREvalTrainer (class in ts1.models.pretrained) RRRReadout (class in ts2.models.single_session) RRRReadoutWithISI (class in ts2.models.single_session) RRRSingleTaskPretrain (class in pretrain.models) run_seed (ts3.Extractor property) (ts3.models.inductive.NEMOExtractor property) (ts3.models.inductive.NuCLRExtractor property) (ts3.models.transductive.TransductiveExtractor property) S save_if_best() (core.trainer.BaseTrainer method) score() (ibl_bwb_eval.TS3ReadoutSpec method) score_dir() (in module ibl_bwb_eval.scoring.ts1) (in module ibl_bwb_eval.scoring.ts2) (in module ibl_bwb_eval.scoring.ts3) score_file() (in module ibl_bwb_eval.scoring.ts1) (in module ibl_bwb_eval.scoring.ts2) (in module ibl_bwb_eval.scoring.ts3) self_attn() (in module core.nn) SEQUENCE (ibl_bwb_eval.TargetResolution attribute) SessionBatchSampler (class in core.samplers) set_epoch() (core.samplers.DistributedSamplerWrapper method) (core.samplers.SessionBatchSampler method) set_factors() (ts3.models.supervised.LossFeedbackSampler method) setup() (core.trainer.BaseTrainer method) (pretrain.models.MtMPretrain method) (pretrain.models.NDT2Pretrain method) (pretrain.models.NDTStitchPretrain method) (pretrain.models.NEDSPretrain method) (pretrain.models.NEMOPretrain method) (pretrain.models.NuCLRPretrain method) (pretrain.models.POSSMMultitaskPretrain method) (pretrain.models.POSSMSingleTaskPretrain method) (pretrain.models.POYOPlusMultitaskPretrain method) (pretrain.models.POYOSingleTaskPretrain method) (pretrain.models.RRRSingleTaskPretrain method) (ts1.models.single_session.CEBRAEvalTrainer method) (ts1.TS1EvalTrainer method) (ts2.models.single_session.NDTEvalTrainer method) (ts2.TS2EvalTrainer method) (ts3.Extractor method) (ts3.models.inductive.NEMOExtractor method) (ts3.models.inductive.NuCLRExtractor method) (ts3.models.supervised.LOLCATTrainer method) (ts3.models.transductive.POSSMCalibrateTrainer method) (ts3.models.transductive.POYOPlusCalibrateTrainer method) (ts3.models.transductive.TransductiveExtractor method) setup_finetuning() (ts1.TS1EvalTrainer method) (ts2.TS2EvalTrainer method) setup_model() (ts1.models.pretrained.MtMEvalTrainer method) (ts1.models.pretrained.NDT2EvalTrainer method) (ts1.models.pretrained.NDTStitchEvalTrainer method) (ts1.models.pretrained.NEDSEvalTrainer method) (ts1.models.pretrained.POSSMEvalTrainer method) (ts1.models.pretrained.POYOEvalTrainer method) (ts1.models.single_session.CEBRAEvalTrainer method) (ts1.TS1EvalTrainer method) (ts2.models.pretrained.MtMEvalTrainer method) (ts2.models.pretrained.NDTStitchEvalTrainer method) (ts2.TS2EvalTrainer method) setup_optimizers() (ts1.TS1EvalTrainer method) (ts2.models.single_session.StatBaselineTrainer method) (ts2.TS2EvalTrainer method) Shrinkage (class in ts2.models.single_session) StatBaseline (class in ts2.models.single_session) StatBaselineTrainer (class in ts2.models.single_session) step() (ts3.models.supervised.LossFeedbackSampler method) step_patience() (core.trainer.BaseTrainer method) summarize() (in module ibl_bwb_eval.scoring.ts1) (in module ibl_bwb_eval.scoring.ts2) (in module ibl_bwb_eval.scoring.ts3) T target_resolution (ibl_bwb_eval.TS1ReadoutSpec attribute) TargetResolution (class in ibl_bwb_eval) task_id() (in module ibl_bwb_eval) TCN (class in ts1.models.single_session) TemporalGaussianSmoothing (class in core.transforms) TemporalJittering (class in core.transforms) test() (core.trainer.BaseTrainer method) (pretrain.models.POSSMMultitaskPretrain method) (pretrain.models.POYOPlusMultitaskPretrain method) (ts3.models.supervised.LOLCATTrainer method) tfixup_init_() (in module core.nn) timestamp_key (ibl_bwb_eval.TS1ReadoutSpec attribute) TIMESTEP (ibl_bwb_eval.TargetResolution attribute) TrailingMean (class in ts2.models.single_session) train() (core.trainer.BaseTrainer method) train_epoch() (core.trainer.BaseTrainer method) (pretrain.models.MtMPretrain method) (pretrain.models.NDT2Pretrain method) (pretrain.models.NDTStitchPretrain method) (pretrain.models.NEDSPretrain method) (pretrain.models.NEMOPretrain method) (pretrain.models.NuCLRPretrain method) (pretrain.models.POSSMMultitaskPretrain method) (pretrain.models.POSSMSingleTaskPretrain method) (pretrain.models.POYOPlusMultitaskPretrain method) (pretrain.models.POYOSingleTaskPretrain method) (pretrain.models.RRRSingleTaskPretrain method) (ts1.TS1EvalTrainer method) (ts2.models.single_session.StatBaselineTrainer method) (ts2.TS2EvalTrainer method) (ts3.models.supervised.LOLCATTrainer method) (ts3.models.transductive.POSSMCalibrateTrainer method) (ts3.models.transductive.POYOPlusCalibrateTrainer method) training_step() (ts2.models.pretrained.MtMEvalTrainer method) (ts2.models.pretrained.NDTStitchEvalTrainer method) (ts2.models.single_session.LFADSEvalTrainer method) (ts2.models.single_session.NDTEvalTrainer method) (ts2.TS2EvalTrainer method) TransductiveExtractor (class in ts3.models.transductive) ts1 module ts1.models.pretrained module ts1.models.single_session module TS1EvalTrainer (class in ts1) TS1ReadoutSpec (class in ibl_bwb_eval) TS1TestMixin (class in ts1) ts2 module ts2.models.pretrained module ts2.models.single_session module TS2EvalTrainer (class in ts2) TS2ReadoutSpec (class in ibl_bwb_eval) TS2TestMixin (class in ts2) ts3 module ts3.models.inductive module ts3.models.supervised module ts3.models.transductive module ts3.probes module TS3ReadoutSpec (class in ibl_bwb_eval) U unit_filtering (core.dataset.IBLBrainWideBench2026 property) UnitDropout (class in core.transforms) UnitFilteringInfo (class in core.data) UnitQCPolicy (class in core.dataset) update() (ibl_bwb_eval.metrics.BPS method) (ibl_bwb_eval.metrics.PoissonD2Score method) Us (pretrain.models.RRRDecoder attribute) uses_xformers() (in module core.nn) V V (pretrain.models.RRRDecoder attribute) val_epoch() (core.trainer.BaseTrainer method) (pretrain.models.MtMPretrain method) (pretrain.models.NDT2Pretrain method) (pretrain.models.NDTStitchPretrain method) (pretrain.models.NEDSPretrain method) (pretrain.models.NEMOPretrain method) (pretrain.models.NuCLRPretrain method) (pretrain.models.POSSMMultitaskPretrain method) (pretrain.models.POSSMSingleTaskPretrain method) (pretrain.models.POYOPlusMultitaskPretrain method) (pretrain.models.POYOSingleTaskPretrain method) (pretrain.models.RRRSingleTaskPretrain method) (ts1.TS1EvalTrainer method) (ts2.models.single_session.StatBaselineTrainer method) (ts2.TS2EvalTrainer method) (ts3.models.supervised.LOLCATTrainer method) validate_attn_impl() (in module core.nn) value_key (ibl_bwb_eval.TS1ReadoutSpec attribute) VariableContextMixin (class in core.dataset) version_of() (in module ibl_bwb_eval.predictions) W WeightedMSELoss (class in core.nn.loss) WholeSessionSpikeDataset (class in core.dataset) WVFEncoder (class in pretrain.models)
