Software
torch_brain
byARNI
A PyTorch-friendly open-source library for learning from Neural Data. Helps you write efficient training pipelines and models.
torch_brain is an end-to-end framework for building deep learning models and training pipelines for neuroscience. It pairs a lightweight, time-based data format (plus tools to preprocess existing neural datasets into it) with PyTorch-compatible building blocks: datasets, samplers, nn.Modules, and models.
Features
- Lazy, on-demand data loading that reads only the time-slices and attributes you request
- Advanced samplers for arbitrary on-the-fly slicing of recordings
- Multi-recording training across heterogeneous datasets
- Support for arbitrary neural and behavioral modalities
- Flexible collation strategies, including chaining and padding
