Installation

torch_brain is available for Python >= 3.10.

To install the package, run:

pip install torch_brain

To install the package from source, run:

git clone https://github.com/neuro-galaxy/torch_brain.git
cd torch_brain
pip install -e .