Prefix functions in lax.py that shouldn't be visible in documentation.
Prefix functions in lax.py that shouldn't be visible in documentation.

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