search

HazyResearch / zoology

Understand and test language model architectures on synthetic tasks.
Apache License 2.0
164 stars 28 forks source link

Added wkv_cuda.cu and wkv_op.cpp files for RWKV #13

Closed guangyusong closed 10 months ago

guangyusong commented 10 months ago

Added wkv_cuda.cu and wkv_op.cpp files for RWKV