Skip to content

Commit 4676238

Browse files
committed
Change title.
1 parent 7e1d356 commit 4676238

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

docs/source/notebooks/GLM-hierarchical-jax.ipynb

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,11 @@
44
"cell_type": "markdown",
55
"metadata": {},
66
"source": [
7-
"# PyMC3-JAX for >20x faster sampling\n",
7+
"# Using JAX for faster sampling\n",
88
"\n",
9-
"Thomas Wiecki\n",
9+
"(c) Thomas Wiecki, 2020\n",
10+
"\n",
11+
"*Note: These samplers are still experimental.*\n",
1012
"\n",
1113
"Using the new Theano JAX linker that Brandon Willard has developed, we can compile PyMC3 models to JAX without any change to the PyMC3 code base or any user-level code changes. The way this works is that we take our Theano graph built by PyMC3 and then translate it to JAX primitives. \n",
1214
"\n",

0 commit comments

Comments
 (0)