Add jit decorators to most functions in jax.scipy.linalg. (#1741)
Add jit decorators to most functions in jax.scipy.linalg. (#1741)


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