doc: partial class not needed in code snippet in README.md
doc: partial class not needed in code snippet in README.md
tensorflow/probability,google/jax,google/jax,google/jax,google/jax,tensorflow/probability