You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Add Stan PPL integration to use Stan models with Blackjax inference algorithms
With the BridgeStan library, we can efficiently access log density and gradient evaluations of Stan models. Following the Blackjax documentation, we can then use BridgeStan with custom gradients and JAX callbacks. Note that Stan, not JAX, would be computing gradients that is then used by Blackjax algorithms.
For a somewhat complete example see my small example repo and another example here. Would be open to attempting a PR if there is interest.
The text was updated successfully, but these errors were encountered:
Add Stan PPL integration to use Stan models with Blackjax inference algorithms
With the BridgeStan library, we can efficiently access log density and gradient evaluations of Stan models. Following the Blackjax documentation, we can then use BridgeStan with custom gradients and JAX callbacks. Note that Stan, not JAX, would be computing gradients that is then used by Blackjax algorithms.
For a somewhat complete example see my small example repo and another example here. Would be open to attempting a PR if there is interest.
The text was updated successfully, but these errors were encountered: