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
Copyright © 2025, neuro-galaxy Team
Made with Sphinx and @pradyunsg's Furo