Made small adjustments for basic tensorflow 2 migration. However, incompatibilities between Edward2, Tensorflow probability [>=0.80], and TF2 are hindering progress.