Error: Attempting to perform BLAS operation using StreamExecutor without BLAS support

[tonygracious]

When I am using jax.scipy.optimize to minimize a loss function I am getting following error. Can anyone explain me what this error means and how can I solve this?

Traceback (most recent call last):
File "/home/tony/hp_jax/hawkes.py", line 105, in
result = jsp_opt.minimize(neg_loglikelihood,
File "/home/tony/miniconda3/lib/python3.8/site-packages/jax/_src/scipy/optimize/minimize.py", line 103, in minimize
results = minimize_bfgs(fun_with_args, x0, **options)
File "/home/tony/miniconda3/lib/python3.8/site-packages/jax/_src/scipy/optimize/bfgs.py", line 164, in minimize_bfgs
state = lax.while_loop(cond_fun, body_fun, state)
jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Attempting to perform BLAS operation using StreamExecutor without BLAS support

The code is working fine in colab Hawkes Process