PyPI page
Home page
Author:
None
Summary:
Linear solvers and eigendecomposition for AWS Trainium via NKI
Latest version:
0.3.0
Required dependencies:
numpy
|
torch
Optional dependencies:
neuronxcc
|
pytest
|
pytest-benchmark
|
scipy
|
torch-neuronx
Downloads last day:
4
Downloads last week:
8
Downloads last month:
149