tfp.experimental.AutoCompositeTensor

Recommended base class for @auto_composite_tensor-ified classes.

See details in tfp.experimental.auto_composite_tensor description.