PyPI page
Home page
Author:
triadastra
Summary:
N-dimensional models for PyTorch: SSMs, RNNs, transformers and convolutions over arbitrary lattices
Latest version:
0.3.1
Required dependencies:
torch
Optional dependencies:
einops
|
flash-linear-attention
|
hydra-core
|
hypothesis
|
mamba-ssm
|
mypy
|
numpy
|
omegaconf
|
pytest
|
pytest-cov
|
pyyaml
|
ruff
|
safetensors
|
scipy
Downloads last day:
0
Downloads last week:
10
Downloads last month:
25
7-day smoothing