PyPI page
Home page
Author:
Scott Friedman
Summary:
Scientific computing suite for AWS Trainium via NKI (FFT, BLAS, RAND, SOLVER, SPARSE, TENSOR)
Latest version:
0.1.0
Optional dependencies:
mkdocs
|
mkdocs-material
|
mkdocs-monorepo-plugin
|
neuronxcc
|
pytest
|
pytest-benchmark
|
torch-neuronx
|
trnblas
|
trnfft
|
trnrand
|
trnsolver
|
trnsparse
|
trntensor
Downloads last day:
0
Downloads last week:
1
Downloads last month:
8
7-day smoothing