Web: https://www.reddit.com/r/MachineLearning/comments/veem7o/r_rwkv2_430m_release_a_parallelizable_rnn_with/

June 17, 2022, 2:35 p.m. | /u/bo_peng

Machine Learning reddit.com

Hi everyone. I posted about my RWKV-2 RNN 1 month ago (thanks for the upvote):


And I have finished the training of a RWKV-2 430M (L24-D1024) on the Pile. It's confirmed that a pure RNN without attention can reach transformer-level LM performance:


The maths behind RWKV-2:


You can download the params & fine-tuning code here:



Now I am training a RWKV-2 1.5B (L24-D2048) which is expected to finish in 2 months :)


attention machinelearning performance rnn transformer

Machine Learning Researcher - Saalfeld Lab

@ Howard Hughes Medical Institute - Chevy Chase, MD | Ashburn, Virginia

Project Director, Machine Learning in US Health

@ ideas42.org | Remote, US

Data Science Intern

@ NannyML | Remote

Machine Learning Engineer NLP/Speech

@ Play.ht | Remote

Research Scientist, 3D Reconstruction

@ Yembo | Remote, US

Clinical Assistant or Associate Professor of Management Science and Systems

@ University at Buffalo | Buffalo, NY