How to save a tfp.distributions.JointDistributionNamed for reuse

[statmike]

Is there a way to save a tfp.distributions.JointDistributionNamed for import in other models?

[jcalifornia]

I created a related issue here: #1494 that is related. What I have done before is write a getstate where I convert all the tf.Tensor variables to numpy and a setstate where I cast the numpy to tf.Tensor and assign to variables.