TorchBrain documentation

torch_brain is a library for designing and training deep learning models on neural data. It provides PyTorch-compatible interfaces like Dataset, Samplers, nn.Modules, and more to help you write your very own models and training scripts.

If you encounter any bugs or have feature requests, please submit them to our GitHub Issues page.

---

• User Guide
  • Installation
  • Sampling

• API Reference
  • torch_brain.dataset
  • torch_brain.data
  • torch_brain.data.sampler
  • torch_brain.transforms
  • torch_brain.nn
  • torch_brain.models
  • torch_brain.registry
  • torch_brain.utils