lnccbrown / HSSM

Development of HSSM package
Other
82 stars 11 forks source link

Fix sampler specification #520

Closed digicosmos86 closed 3 months ago

digicosmos86 commented 4 months ago

In Bambi 0.14.0, all JAX-based sampling is now done via bayeux, which breaks our sampler specification. For the moment, we are sticking to JAX-based sampling via pymc, through the nuts_sampler argument of pm.sample