create bayesian logistic regression notebook demo (#103)

* Add bayesian logistic regression with RMH * Change the type of PRNGKeys to DeviceArray
blackjax-devs · Dec 22, 2021 · 6f728a9 · 6f728a9
1 parent f2ccaf7
commit 6f728a9
Show file tree

Hide file tree

Showing 2 changed files with 417 additions and 1 deletion.
diff --git a/blackjax/types.py b/blackjax/types.py
@@ -9,4 +9,4 @@
 # Once recursive typing is fully supported (https://github.com/python/mypy/issues/731), we can uncomment the line below.
 # PyTree = Union[Array, Iterable["PyTree"], Mapping[Any, "PyTree"]]
 
-PRNGKey = Array
+PRNGKey = jnp.ndarray