SeanSong-amd

Pinned Repositories

causal-conv1d
Add support for ROCm
Language:Cuda01
causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
Language:Cuda258 4 1752
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python219 26 21650
rocm-blogs
Language:Jupyter Notebook26 11 46
causal-conv1d
Causal depthwise conv1d in ROCm, with a PyTorch interface
Language:Cuda2 0 00

SeanSong-amd's Repositories

SeanSong-amd/causal-conv1d
Causal depthwise conv1d in ROCm, with a PyTorch interface
Language:Cuda2 0 00