Slopped together to understand a couple of papers better. - View it on GitHub