PyPI page
Home page
Author:
None
Summary:
High-performance ML primitives, applications, and informative cost API — Triton + CuteDSL kernels for NVIDIA GPUs.
Latest version:
0.2.0
Required dependencies:
numba
|
numpy
|
nvidia-cutlass-dsl
|
torch
|
tqdm
|
triton
Optional dependencies:
matplotlib
|
pytest
|
scikit-learn
|
tqdm
Downloads last day:
88
Downloads last week:
451
Downloads last month:
2,848