Pinned Repositories
causal-conv1d
Add support for ROCm
causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
rocm-blogs
causal-conv1d
Causal depthwise conv1d in ROCm, with a PyTorch interface
SeanSong-amd's Repositories
SeanSong-amd/causal-conv1d
Causal depthwise conv1d in ROCm, with a PyTorch interface