Metadata-Version: 2.4
Name: neuraltrain
Version: 0.0.3
Summary: A lightweight deep learning library for learning on M/EEG and other neuroimaging modalities.
Author: Meta FAIR
License-Expression: MIT
Project-URL: Source, https://github.com/facebookresearch/neuroai
Project-URL: Tracker, https://github.com/facebookresearch/neuroai/issues
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.12
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: pandas>=2.0.1
Requires-Dist: numpy>=2.1
Requires-Dist: scikit-learn>=0.21.2
Requires-Dist: torch>=2.5.1
Requires-Dist: torchvision>=0.20.1
Requires-Dist: packaging
Requires-Dist: pydantic>=2.5.0
Requires-Dist: torchmetrics>=1.1.2
Provides-Extra: models
Requires-Dist: braindecode>=1.4.0; extra == "models"
Requires-Dist: x_transformers>=1.27.20; extra == "models"
Requires-Dist: dalle2_pytorch; extra == "models"
Requires-Dist: vit_pytorch; extra == "models"
Provides-Extra: metrics
Requires-Dist: openai-clip; extra == "metrics"
Provides-Extra: lightning
Requires-Dist: lightning>=2.0.8; extra == "lightning"
Requires-Dist: wandb>=0.15.11; extra == "lightning"
Provides-Extra: dev
Requires-Dist: pytest>=4.3.0; extra == "dev"
Requires-Dist: pytest-markdown-docs; extra == "dev"
Requires-Dist: ruff>=0.15; extra == "dev"
Requires-Dist: pre-commit>=1.15.2; extra == "dev"
Requires-Dist: mypy>=0.971; extra == "dev"
Requires-Dist: pandas-stubs; extra == "dev"
Requires-Dist: types-PyYAML; extra == "dev"
Requires-Dist: types-Pillow; extra == "dev"
Requires-Dist: types-pywin32; extra == "dev"
Requires-Dist: types-requests; extra == "dev"
Requires-Dist: types-setuptools; extra == "dev"
Requires-Dist: types-tabulate; extra == "dev"
Requires-Dist: types-tqdm; extra == "dev"
Dynamic: license-file

# 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

```bash
pip install neuraltrain
```

## Citation

Citation coming soon.

## License

This project is licensed under the MIT License. See [LICENSE](LICENSE) for details.
