This package allows you to use Slang as a language to write PyTorch kernels.
Slang user guide: https://shader-slang.com/slang/user-guide/
slangtorch
documentation: https://shader-slang.org/slang/docs/user-guide/a1-02-slangpy
Note: Running slangtorch
requires a CUDA-capable GPU.
- Install the CUDA Toolkit v12.0 or later
- Install PyTorch v2.x with CUDA support from https://pytorch.org/.
pip install slangtorch
to install slangtorch.