Wheelodex » Projects » pytorch-brain

pytorch-brain

View on PyPI — Reverse Dependencies (0)

0.1.0 pytorch_brain-0.1.0-py3-none-any.whl

Wheel Details

Project: pytorch-brain
Version: 0.1.0
Filename: pytorch_brain-0.1.0-py3-none-any.whl
Download: [link]
Size: 65860
MD5: b48311986a115dabb5504c878130e933
SHA256: a32f87abf633dcfbe853ae59f35525e860c63ea36cf1a93688c7a49028ec418b
Uploaded: 2025-03-26 20:32:36 +0000

dist-info

METADATA · WHEEL · RECORD · top_level.txt

METADATA

Metadata-Version: 2.4
Name: pytorch_brain
Version: 0.1.0
Summary: A package for deep learning models for neuroscience
Author-Email: Mehdi Azabou <mehdiazabou[at]gmail.com>, Vinam Arora <vinam[at]gatech.edu>
Project-Url: Homepage, https://github.com/neuro-galaxy/torch_brain
Project-Url: Issues, https://github.com/neuro-galaxy/torch_brain/issues
Project-Url: Documentation, https://torch-brain.readthedocs.io/en/latest/
License: Apache-2.0
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.8
Requires-Dist: temporaldata (>=0.1.3)
Requires-Dist: numpy
Requires-Dist: torch (~=2.0)
Requires-Dist: einops (~=0.6.0)
Requires-Dist: hydra-core (~=1.3.2)
Requires-Dist: torchtyping (~=0.1)
Requires-Dist: torchmetrics (>=1.6.0)
Requires-Dist: pydantic (~=2.0)
Requires-Dist: rich
Requires-Dist: pytest; extra == "dev"
Requires-Dist: black (==24.2.0); extra == "dev"
Requires-Dist: pre-commit (>=3.5.0); extra == "dev"
Requires-Dist: flake8; extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
Dynamic: license-file
License-File: LICENSE.txt
[Description omitted; length: 2917 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (78.1.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
pytorch_brain-0.1.0.dist-info/licenses/LICENSE.txt sha256=b_25GJNOtkXBuGSf0grLcbRGLFkywH12a-gq_zWe6dY 11341
torch_brain/__init__.py sha256=tai6QYDqk1wu3tjscMvMU4RmPDZDPmDKLLfeect5taY 225
torch_brain/optim.py sha256=AlRRyLN6ODo0yNpZ2b2wNQ2oOTeuw8sjSrkfC0r8Hg4 6874
torch_brain/registry.py sha256=H8NRqSuicjyc-WTq7oV2DXgvSr88l8KcJLdA9Zhjkf8 6976
torch_brain/data/__init__.py sha256=_Ll8imKh8qUg3-bBz3qEfOCPUZSW8yHZKmJbVRNxMqM 136
torch_brain/data/collate.py sha256=frMMJhn9fm9z-Af__Md-5JZ10y3ne3qPKsc1mkJSaWw 7867
torch_brain/data/dataset.py sha256=8g65YPmcGeK4I9QvHZshjgY6uh5rpGLuyTMv3ZQlnAQ 20235
torch_brain/data/sampler.py sha256=JUMwk-FLQBlEBTJj_xusz1PLibNEuR-NIxl8yncqgf4 19843
torch_brain/models/__init__.py sha256=tSOH15y_XpuRmzwAygrKJUjd_VsL6pa4PAIzJY8a7Po 64
torch_brain/models/poyo.py sha256=7kbr2JZA0CpKIDpFR4UI98BghyeHlvl0QBnQMeXKQDY 14489
torch_brain/models/poyo_plus.py sha256=9SJfPKXJxUlM_cRwDF2IvNBCYOd3d4fC9AgfGEhYu8Q 14273
torch_brain/nn/__init__.py sha256=1ZzeTeGAzrY5EnDyNqrI2iRw4jeapRF04vDKkO3UmrE 452
torch_brain/nn/embedding.py sha256=QsuUGV688Bx5KTGYrZiKSbyNCJKfi0S-qyYpKsPWgYI 1280
torch_brain/nn/feedforward.py sha256=3KBqq-JeQbMOAZdRsn_REVONCDfk7s31J6FzjByzjcE 972
torch_brain/nn/infinite_vocab_embedding.py sha256=JBN4qxbPGjJrKqHErDe5n0Vxllv8kax4JLbGudu8m1s 17041
torch_brain/nn/loss.py sha256=nFUiVZT59UAs5YnfeQ1x3yJ5Vli3oH2jiXW3xw4eJww 3865
torch_brain/nn/multitask_readout.py sha256=9YJPCE50AlnQehu-gVm_APbPj3UWhaY8GQC5GKt-yOw 10278
torch_brain/nn/rotary_attention.py sha256=7X7VXnnj_vjDOvR9WiJAE2ZvAe4FJolMrygbVtyUxvU 18209
torch_brain/nn/rotary_embedding.py sha256=O24q7qzJmql0BrYph9eA8NydujYcMFhWLQyBDJpgaaE 2319
torch_brain/transforms/__init__.py sha256=t2W7I3HAhi9gwLCgaf4sfWTSkuWes6l23X70-6PVvAo 311
torch_brain/transforms/container.py sha256=uEdQ2reF46oVtGuMLxHJFY4oRbrY4KQy9RsKUGvxZM8 2817
torch_brain/transforms/output_sampler.py sha256=F23qOa8_6Xbrskq1G4IMIbA9pM_muCofAkPoIxK2yQ8 629
torch_brain/transforms/random_crop.py sha256=A_bhAsvSlUxf_qxKfWZtnM30H3VcnwgxOB1oeJZAVOE 389
torch_brain/transforms/random_time_scaling.py sha256=tF67iIPlucK4hAG0ynFW1Xv-72tlNcTAC0zzLrMQFMk 2381
torch_brain/transforms/unit_dropout.py sha256=rFDroTQRNPcEPp1W9XWEUaw92HeZ58RQqQ0Dx0Ze5uA 6859
torch_brain/transforms/unit_filter.py sha256=xTn4WVQzVIcl2DanqiiQ2CrLAqlUBEJ75CMov3pAZjE 3156
torch_brain/utils/__init__.py sha256=PIKWXo9waXD50eJ1Ws38lgfW4bd6CowVZ1f29nnF_Ls 308
torch_brain/utils/binning.py sha256=JpsigGZJUk1vhoVLKl-WbFGf68YSq8CxyW2_jAkRjWM 1719
torch_brain/utils/callbacks.py sha256=gLr0aztoVwkVAkVX3gKnhd1EcC5Me2mzKn8sNYr0yto 4055
torch_brain/utils/gradient_rescale.py sha256=ABWGRajJc_XrdiLkPQSe5a29G0asvWe5cYcACxyma6c 1720
torch_brain/utils/readout.py sha256=Zfa7x6d1OkwsF2MHZL1hN1ItrsPqMKxVoAMX_ZTrMD4 3089
torch_brain/utils/seed_everything.py sha256=5yOhV0E1dAqv5HYm4_rHAbqfIrxeJDFEyjN_zqyyKu0 602
torch_brain/utils/sinusoidal_encoding.py sha256=tDX_npVCFI0fTWJkjO2XkIR6tDmie6ObtWOdG0odocY 907
torch_brain/utils/stitcher.py sha256=DH4nqrew5Umjn0aG_qZaqlTyxA12M6Psl4i1BFc9S_w 18996
torch_brain/utils/tokenizers.py sha256=s4NVtl7Z5SZGkfX_7sW76dRmx6C6TfB4UJ3_iO4XfxI 2019
torch_brain/utils/weights.py sha256=DGFun9pblJZzI78ByEZ3taHLAdXXQrQkJqTdrR4gss0 2853
pytorch_brain-0.1.0.dist-info/METADATA sha256=ccIOEtz1pySt-WNE4VH884km20413VvogAcnU9ojT08 4071
pytorch_brain-0.1.0.dist-info/WHEEL sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0 91
pytorch_brain-0.1.0.dist-info/top_level.txt sha256=wSKTR8IvyGIOHXslnf_L-C5Brj6bKHBak2a1UyMgOAE 12
pytorch_brain-0.1.0.dist-info/RECORD — —

top_level.txt

torch_brain