Thank you! This works, but the sampling seems to proceed a bit slow at 3/s. This is with GPU, on CPU it’s more like 1-2 s/draw (and I’m still getting the blas warning). Not sure if I’m missing any settings? Metropolis was sampling at 15-20/s. Still running the logistic reg with some 30K observations. I’ve added the output of np.show_config() below.
Auto-assigning NUTS sampler...
Initializing NUTS using adapt_diag...
Sequential sampling (2 chains in 1 job)
NUTS: [educ, hours, I(age ** 2), age, sex[T. Male], Intercept]
11%|█ | 1173/11000 [07:18<46:54, 3.49it/s]
blas_mkl_info:
NOT AVAILABLE
blis_info:
NOT AVAILABLE
openblas_info:
libraries = ['openblas', 'openblas']
library_dirs = ['/usr/local/lib']
language = c
define_macros = [('HAVE_CBLAS', None)]
blas_opt_info:
libraries = ['openblas', 'openblas']
library_dirs = ['/usr/local/lib']
language = c
define_macros = [('HAVE_CBLAS', None)]
lapack_mkl_info:
NOT AVAILABLE
openblas_lapack_info:
libraries = ['openblas', 'openblas']
library_dirs = ['/usr/local/lib']
language = c
define_macros = [('HAVE_CBLAS', None)]
lapack_opt_info:
libraries = ['openblas', 'openblas']
library_dirs = ['/usr/local/lib']
language = c
define_macros = [('HAVE_CBLAS', None)]