Implementation of https://srush.github.io/annotated-s4 - View it on GitHub