Skip to main content
Back to top
Ctrl
+
K
TorchBrain
Install TorchBrain
User Guide
CLI Reference
API Reference
Search
Ctrl
+
K
System Settings
Light
Dark
GitHub
Discord
PyPI
Search
Ctrl
+
K
Install TorchBrain
User Guide
CLI Reference
API Reference
System Settings
Light
Dark
GitHub
Discord
PyPI
Collapse Sidebar
Expand Sidebar
Section Navigation
torch_brain.batching
chain
collate
pad
pad8
pad2d
pad2d8
track_batch
track_mask
track_mask8
track_mask2d
track_mask2d8
torch_brain.data
ArrayDict
LazyArrayDict
IrregularTimeSeries
LazyIrregularTimeSeries
RegularTimeSeries
LazyRegularTimeSeries
Interval
LazyInterval
Data
BrainsetDescription
DeviceDescription
SessionDescription
SubjectDescription
concat
serialize_fn_map
torch_brain.datasets
Dataset
DatasetIndex
NestedDataset
OpenNeuroDataset
SpikingDatasetMixin
CalciumImagingDatasetMixin
MultiChannelDatasetMixin
PerichMillerPopulation2018
PeiPandarinathNLB2021
FlintSlutzkyAccurate2012
ChurchlandShenoyNeural2012
OdohertySabesNonhuman2017
VollanMoserAlternating2025
ShiraziHBNR1DS005505
AllenVisualCodingOphys2016
Neuroprobe2025
KochiVisualNamingDS006914
KlinzingSleepDS005555
KempSleepEDF2013
torch_brain.models
POYO
torch_brain.nn
Embedding
InfiniteVocabEmbedding
RotaryTimeEmbedding
SinusoidalTimeEmbedding
RotaryCrossAttention
RotarySelfAttention
torch_brain.pipeline
BrainsetPipeline
torch_brain.pipeline.openneuro
OpenNeuroPipeline
torch_brain.samplers
RandomFixedWindowSampler
SequentialFixedWindowSampler
TrialSampler
DistributedEvaluationSamplerWrapper
DistributedStitchingFixedWindowSampler
torch_brain.transforms
Compose
RandomChoice
ConditionalChoice
UnitDropout
TriangleDistribution
RandomTimeScaling
RandomOutputSampler
RandomCrop
BinSpikes
UnitFilter
UnitFilterById
torch_brain.utils
create_linspace_latent_tokens
create_start_end_unit_tokens
resolve_weights_based_on_interval_membership
isin_interval
np_string_prefix
bin_spikes
torch_brain.utils.bids
fetch_eeg_recordings
fetch_ieeg_recordings
group_recordings_by_entity
check_eeg_recording_files_exist
check_ieeg_recording_files_exist
build_bids_path
load_json_sidecar
load_participants_tsv
get_subject_info
torch_brain.utils.dandi
extract_subject_from_nwb
extract_spikes_from_nwbfile
download_file
get_nwb_asset_list
torch_brain.utils.mne
extract_measurement_date
concatenate_recordings
extract_signal
extract_channels
torch_brain.utils.openneuro
OPENNEURO_S3_BUCKET
construct_s3_url_from_path
download_dataset_description
download_recording
fetch_latest_snapshot_tag
fetch_all_filenames
fetch_participants_tsv
fetch_species
torch_brain.utils.s3
get_cached_s3_client
get_object_list
download_prefix
download_prefix_from_url
torch_brain.utils.signal
downsample_wideband
extract_bands
cube_to_long
torch_brain.utils.split
generate_stratified_folds
generate_string_kfold_assignment
torch_brain.utils.stitcher
stitch
API Reference
torch_brain.data
serialize_fn_map
serialize_fn_map
#
dict() -> new empty dictionary