Comment Comment by darkbatman By looking at the paper, memory needed per layer seems to be higher than transformer architecture. Pretty sure that would be blowing up the vram of gpu at scale. parent darkbatman 1 year ago