LAPACK 3.6 and below requires more workspace that the newer versions
LAPACK 3.6 and below requires more workspace that the newer versions

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