All modules for which code is available

• torch_brain.batching.collate
• torch_brain.data.dataset
• torch_brain.dataset.dataset
• torch_brain.dataset.mixins
• torch_brain.dataset.nested
• torch_brain.models.poyo
• torch_brain.nn.embedding
• torch_brain.nn.infinite_vocab_embedding
• torch_brain.nn.position_embeddings
• torch_brain.nn.rotary_attention
• torch_brain.samplers.distributed_evaluation_sampler
• torch_brain.samplers.distributed_stitching_fixed_window
• torch_brain.samplers.random_fixed_window
• torch_brain.samplers.sequential_fixed_window
• torch_brain.samplers.trial_sampler
• 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.stitcher
• torch_brain.utils.tokenizers
• torch_brain.utils.weights