rocm-port

Pinned Repositories

causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
Language:Cuda00
hyper-attn
Language:Python00
LWM
Language:Python00
mamba-rocm
Porting Mamba on ROCm hardware
Language:Python20
mamba.py
A simple and efficient Mamba implementation in PyTorch and MLX.
Language:Python00

rocm-port's Repositories

rocm-port/mamba-rocm
Porting Mamba on ROCm hardware
Language:Python20
rocm-port/causal-conv1d
Causal depthwise conv1d in CUDA, with a PyTorch interface
Language:Cuda00
rocm-port/hyper-attn
Language:Python00
rocm-port/LWM
Language:Python00
rocm-port/mamba.py
A simple and efficient Mamba implementation in PyTorch and MLX.
Language:Python00