jax

f02d1223 - [shape_poly] linalg.cholesky: add support for shape polymorphism with native lowering on CPU