Make the random number generator abstract.
Add support for tf-random.