PyPI page
Home page
Author:
Scott Friedman
Summary:
BLAS operations for AWS Trainium via NKI
Latest version:
0.5.4
Required dependencies:
numpy
|
torch
Optional dependencies:
neuronxcc
|
nki
|
pre-commit
|
pyscf
|
pytest
|
pytest-benchmark
|
pytest-cov
|
ruff
|
scipy
|
torch-neuronx
Downloads last day:
0
Downloads last week:
19
Downloads last month:
3
7-day smoothing