All modules for which code is available

• torch_brain.data.collate
• torch_brain.data.dataset
• torch_brain.data.sampler
• torch_brain.dataset.dataset
• torch_brain.dataset.mixins
• torch_brain.dataset.nested
• torch_brain.models.calcium_poyo_plus
• torch_brain.models.poyo
• torch_brain.models.poyo_plus
• torch_brain.nn.embedding
• torch_brain.nn.infinite_vocab_embedding
• torch_brain.nn.loss
• torch_brain.nn.multitask_readout
• torch_brain.nn.position_embeddings
• torch_brain.nn.rotary_attention
• torch_brain.registry
• torch_brain.transforms.bin_spikes
• torch_brain.transforms.container
• torch_brain.transforms.output_sampler
• torch_brain.transforms.random_crop
• torch_brain.transforms.random_time_scaling
• torch_brain.transforms.unit_dropout
• torch_brain.transforms.unit_filter
• torch_brain.utils.binning
• torch_brain.utils.misc
• torch_brain.utils.readout
• torch_brain.utils.seed_everything
• torch_brain.utils.stitcher
• torch_brain.utils.tokenizers
• torch_brain.utils.weights