4 ms·

Hi, thanks for this fascinating demo. I am wondering how this architecture optimizes for the softmax part.