pypi.org : titans-pytorch : 0.4.0
Titans
Registry -
Documentation -
Download -
JSON
Integrity: sha256-bf470b93ab01fd37c9fe... -
purl: pkg:pypi/[email protected]
Published:
Indexed:
Loading...
Readme
Loading...
Dependencies
- adam-atan2-pytorch >=0.1.18 extra == "examples" (optional)
- wandb * extra == "examples" (optional)
- pytest * extra == "test" (optional)
- accelerated-scan >=0.2.0
- axial-positional-embedding >=0.3.10
- einops >=0.8.0
- einx >=0.3.0
- hyper-connections >=0.1.11
- ninja *
- rotary-embedding-torch *
- tensordict *
- torch >=2.2
- tqdm *
- x-transformers *