PyPI page
Author:
Emily Shepperd
Summary:
Call CuTe DSL kernels from JAX via TVM-FFI target registration
Latest version:
0.1.5
Required dependencies:
jax
|
jax-tvm-ffi
|
nvidia-cutlass-dsl
Optional dependencies:
nvidia-cutlass-dsl
|
redis
Downloads last day:
2
Downloads last week:
36
Downloads last month:
162
7-day smoothing