The associated library by the authors, written in Haskell, can be found at https://github.com/cpeikert

I’ve not had chance to work through the material but looks like it might be fun!