Tensor flow probability

[jstac]

Tensor flow probability now works on JAX: https://www.tensorflow.org/probability/examples/TensorFlow_Probability_on_JAX

It includes routines for Monte Carlo and multivariate optimization.

Are there any lectures in https://python.quantecon.org and https://python-advanced.quantecon.org/intro.html that we can now bring over to this repo as a result?

[jstac]

@chappiewuzefan , could you please investigate?

[chappiewuzefan]

Sure