tfp.experimental.substrates.jax.util.TransformedVariable

View source on GitHub