Open
Description
Catalyst support for certain JAX functions relies on a BLAS runtime library being available, specifically the OpenBLAS library shipped with SciPy. The relevant source code is located in the Python compiler driver:
catalyst/frontend/catalyst/compiler.py
Lines 289 to 304 in 72698d8
This code snippet is brittle and susceptible to different packaging configurations (for instance the assumption of the folder locations relative to the scipy package location.
At minimum we should raise a user friendly error message that there is a problem with the scipy installation.
Ideally, we would also find ways to make this more robust on the various systems, and also consider conda installations.