Alright, I wrote a colab to demonstrate how a separate API of 1) transform distribution, and 2) transforming bounded Variables should solve this (the example is in TFP): https://colab.research.google.com/drive/1Swagke96szpczDD7Iflf958B7CNRwjXV?usp=sharing