TPU RWKV Implementations This repo has a port of RWKV to jax that can run on tpus. This work was supported by TRC.