mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix List rendering in RTD for primitives.ipynb (#1501)

Browse Source 
Colab doesn't require a newline before unordered list in Markdown; RTD
does.
This commit is contained in:
Trevor Cai 2019-10-14 18:55:13 +01:00 committed by Stephan Hoyer
parent 16219358e2
commit c50495b272
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/notebooks/How_JAX_primitives_work.ipynb
View File

@ -547,6 +547,7 @@
"JAX first evaluates it abstractly using only the \n",
"shape and type of the arguments. This abstract evaluation serves multiple\n",
"purposes:\n",
"\n",
" * Gets the sequence of JAX primitives that are used in the computation. This \n",
" sequence will be compiled. \n",
" * Computes the shape and type of all vectors and operations used in the computation. \n",
@ -976,6 +977,7 @@
},
"source": [
"TO EXPLAIN: \n",
"\n",
" * Why is JAX using ConcreteArray in square_add_prim? There is no abstract evaluation going on here.\n",
" * Not sure how to explain that multiply_add_prim is invoked with ConcreteValue, yet\n",
" we do not call the multiply_add_abstract_eval.\n",