PyPI page
Home page
Author:
Scott Friedman
Summary:
Tensor contractions for AWS Trainium via NKI
Latest version:
0.3.0
Required dependencies:
numpy
|
torch
Optional dependencies:
mkdocs
|
mkdocs-material
|
pre-commit
|
pytest
|
pytest-benchmark
|
pytest-cov
|
ruff
|
scipy
Downloads last day:
6
Downloads last week:
7
Downloads last month:
7
7-day smoothing