I put out a PR for my optimization work 19 and wrote a bit about JIT and LSTM 37.
So with a bit of analysis of what’s slowing things down and a fair amount of bookkeeping, you can speed up things quite a bit.
Best regards
Thomas