LICENSE.txt
README.md
setup.py
rwkv_ops/__init__.py
rwkv_ops.egg-info/PKG-INFO
rwkv_ops.egg-info/SOURCES.txt
rwkv_ops.egg-info/dependency_links.txt
rwkv_ops.egg-info/requires.txt
rwkv_ops.egg-info/top_level.txt
rwkv_ops/rwkv7_kernel/__init__.py
rwkv_ops/rwkv7_kernel/get_jax_devices_info.py
rwkv_ops/rwkv7_kernel/get_torch_devices_info.py
rwkv_ops/rwkv7_kernel/jax_op.py
rwkv_ops/rwkv7_kernel/native_keras_op.py
rwkv_ops/rwkv7_kernel/torch_op.py
rwkv_ops/rwkv7_kernel/jax_kernel/__init__.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_A_bwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_A_fwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_h_bwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_h_fwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_o_bwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/chunk_o_fwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/cumsum.py
rwkv_ops/rwkv7_kernel/jax_kernel/wy_fast_bwd.py
rwkv_ops/rwkv7_kernel/jax_kernel/wy_fast_fwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/__init__.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_A_bwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_A_fwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_h_bwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_h_fwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_o_bwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/chunk_o_fwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/cumsum.py
rwkv_ops/rwkv7_kernel/torch_kernel/wy_fast_bwd.py
rwkv_ops/rwkv7_kernel/torch_kernel/wy_fast_fwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/__init__.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_A_bwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_A_fwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_h_bwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_h_fwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_o_bwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/chunk_o_fwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/cumsum.py
rwkv_ops/rwkv7_kernel/triton_kernel/utils.py
rwkv_ops/rwkv7_kernel/triton_kernel/wy_fast_bwd.py
rwkv_ops/rwkv7_kernel/triton_kernel/wy_fast_fwd.py