shrew-cuda

CUDA backend for the Shrew tensor engine.

Features:

• Bindings to NVIDIA libraries (cuBLAS, cuDNN).
• Custom CUDA kernels defined in kernels/.
• Efficient data transfer between Host and Device.

Prerequisites

• CUDA Toolkit 12.0+
• NVCC

License

Apache-2.0