Pytorch BABI-MEMNN somewhere

Is there a pytorch version of the BABI-MEMNN notebook somewhere?

Or is someone aware of a good simple implementation somewhere else? I can find a number on Github but they all seem way too complicated.