Skip to main content
Ctrl+K

TorchBrain

  • User Guide
  • API Reference
  • GitHub
  • Discord
  • PyPI
  • User Guide
  • API Reference
  • GitHub
  • Discord
  • PyPI

Section Navigation

  • torch_brain.dataset
    • Dataset
    • DatasetIndex
    • NestedDataset
    • SpikingDatasetMixin
    • CalciumImagingDatasetMixin
    • MultiChannelDatasetMixin
  • torch_brain.data
    • chain
    • collate
    • pad
    • pad8
    • pad2d
    • pad2d8
    • track_batch
    • track_mask
    • track_mask8
    • track_mask2d
    • track_mask2d8
    • Dataset
  • torch_brain.data.sampler
    • RandomFixedWindowSampler
    • SequentialFixedWindowSampler
    • TrialSampler
    • DistributedEvaluationSamplerWrapper
    • DistributedStitchingFixedWindowSampler
  • torch_brain.transforms
    • Compose
    • RandomChoice
    • ConditionalChoice
    • UnitDropout
    • TriangleDistribution
    • RandomTimeScaling
    • RandomOutputSampler
    • RandomCrop
    • BinSpikes
    • UnitFilter
    • UnitFilterById
  • torch_brain.nn
    • Embedding
    • InfiniteVocabEmbedding
    • RotaryTimeEmbedding
    • SinusoidalTimeEmbedding
    • FeedForward
    • RotaryCrossAttention
    • RotarySelfAttention
    • MultitaskReadout
    • prepare_for_multitask_readout
    • Loss
    • MSELoss
    • CrossEntropyLoss
    • MallowDistanceLoss
  • torch_brain.models
    • POYO
    • POYOPlus
    • CalciumPOYOPlus
    • poyo_mp
  • torch_brain.registry
    • DataType
    • ModalitySpec
    • register_modality
    • get_modality_by_id
    • MODALITY_REGISTRY
  • torch_brain.utils
    • stitch
    • seed_everything
    • create_linspace_latent_tokens
    • create_start_end_unit_tokens
    • resolve_weights_based_on_interval_membership
    • isin_interval
    • prepare_for_readout
    • np_string_prefix
    • bin_spikes
  • API Reference
  • torch_brain.models

torch_brain.models#

POYO

POYO model from Azabou et al. 2023, A Unified, Scalable Framework for Neural Population Decoding.

POYOPlus

POYO+ model from Azabou et al. 2025, Multi-session, multi-task neural decoding from distinct cell-types and brain regions.

CalciumPOYOPlus

poyo_mp

previous

MallowDistanceLoss

next

POYO

© Copyright 2026, neuro-galaxy Team.

Created using Sphinx 8.1.3.

Built with the PyData Sphinx Theme 0.18.0.