rwkv_simple
rwkv_simple is an easy-to-use implementation of RWKV-6 (x060). We support
multiple WKV kernels (a Triton-based one from Flash Linear Attention, the official
CUDA kernel, and a pure-PyTorch one).
Flash Linear Attention
We use the WKV6 kernel from the Flash Linear Attention (FLA) project by default, so for best performance, please install FLA directly from its repository:
pip install -U git+https://github.com/sustcsonglin/flash-linear-attention
See: https://github.com/sustcsonglin/flash-linear-attention.
License
Copyright © 2024 Ronsor Labs. Licensed under the Apache License, version 2.0.
1 maintainer
Release files for rwkv-simple 0.1.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| rwkv_simple-0.1.1.tar.gz | 392.4 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| rwkv_simple-0.1.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 783.5 kB
Release files / rwkv_simple-0.1.1.tar.gz
| Download URL | rwkv_simple-0.1.1.tar.gz |
|---|---|
| Size | 392.4 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
08f71a2f2e4c4738621577f839b798352686221f83e56f30b127d4ca37d37550
|
|
BLAKE2b-256 checksum How to use checksums |
5d1466d49d1a10aa22880bccb85754e1a402c716503868f6536e355d6ed047eb
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.0 CPython/3.11.6
|
Release files / rwkv_simple-0.1.1-py3-none-any.whl
| Download URL | rwkv_simple-0.1.1-py3-none-any.whl |
|---|---|
| Size | 391.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
6aa731bc855f405d4ba2a3472960943905681986a4576a48bd86ef604fc813b4
|
|
BLAKE2b-256 checksum How to use checksums |
104c5eb1160c7fa350c7b588a9a7f0e663f24b2917fcedbdd815c537c9ac8261
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.0 CPython/3.11.6
|
