avoid attempt to read literals from the typechecking environment
avoid attempt to read literals from the typechecking environment

tensorflow/probability,tensorflow/probability,google/jax,google/jax,google/jax,google/jax