This was my private research codebase during grad school. After graduation I made it all public - View it on GitHub