Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
torch_brain documentation
Logo
torch_brain documentation

Get Started

  • Installation

Concepts

  • Sampling

Package Reference

  • torch_brain.data
    • Dataset
    • Collate
    • Samplers
  • torch_brain.nn
    • Embeddings Layers
    • Transformer modules
    • Losses
    • Readout Layers
  • torch_brain.models
  • torch_brain.registry
  • torch_brain.optim
  • torch_brain.transforms
  • torch_brain.utils
Back to top

torch_brainΒΆ

PyPI Package GitHub Repository GitHub Issues Discord

torch_brain is a Python library for various deep learning models designed for neuroscience.

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


Get Started

  • Installation

Concepts

  • Sampling

Package Reference

  • torch_brain.data
  • torch_brain.nn
  • torch_brain.models
  • torch_brain.registry
  • torch_brain.optim
  • torch_brain.transforms
  • torch_brain.utils
Next
Installation
Copyright © 2025, neuro-galaxy Team
Made with Sphinx and @pradyunsg's Furo