neuraltrain 0.2.1

A lightweight deep learning library for learning on M/EEG and other neuroimaging modalities.

neuraltrain

A lightweight deep learning library for M/EEG, fMRI, and other neuroimaging modalities.

• PyTorch + Lightning for scalable model training
• Specialized architectures for MEG, EEG, fMRI, and multi-modal data
• Domain-specific losses and metrics
• Multi-GPU training with distributed backends

Install

pip install neuraltrain

Citation

If you use NeuralTrain in your research, please cite A foundation model of vision, audition, and language for in-silico neuroscience:

@article{dAscoli2026TribeV2,
  title={A foundation model of vision, audition, and language for in-silico neuroscience},
  author={d'Ascoli, St{\'e}phane and Rapin, J{\'e}r{\'e}my and Benchetrit, Yohann and Brooks, Teon and Begany, Katelyn and Raugel, Jos{\'e}phine and Banville, Hubert and King, Jean-R{\'e}mi},
  year={2026}
}

License

This project is licensed under the MIT License. See LICENSE for details.