20 Commits

Author SHA1 Message Date
Sharad Vikram
aca9dc6c6f Update custom interpreter tutorial 2022-05-11 11:11:47 -07:00
Jake VanderPlas
dd71e2920b DOC: remove mentions of units 2022-05-04 16:08:44 -07:00
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07:00
George Necula
6a48c60a72 Rename master to main in embedded links.
Tried to avoid the change on external links to repos that
have not yet renamed master.
2021-06-18 10:00:01 +03:00
Sharad Vikram
577c601191 Update references in custom interpreter notebook 2021-04-12 19:00:28 -07:00
Jake VanderPlas
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -08:00
Jake VanderPlas
586b3bb6a9 DOC: regularize notebook metadata 2021-02-22 10:04:42 -08:00
Jake VanderPlas
d85d204897 DOC: change build from nbsphinx to myst 2021-02-16 10:28:39 -08:00
Jake VanderPlas
6eedadc27f DOC: use jupytext to maintain synced markdown versions of notebooks. 2021-02-10 12:23:07 -08:00
Peter Hawkins
3ac809ede3 [JAX] Move jax.util to jax._src_util.
PiperOrigin-RevId: 351234602
2021-01-11 14:21:07 -08:00
Matthew Johnson
09d8ac14de
use fewer internal APIs in custom interp notebook (#4016) 2020-08-10 20:48:03 -07:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks (#3760) 2020-07-15 13:17:38 -07:00
George Necula
abbc70b20a Added type annotations and comments related to partial evaluation.
Introduced two new constructors for PartialVal: unknown and known.
These should make it easier to read the code where we construct
PartialVal:

 * instead of PartialVal((aval, core.unit) we use PartialVal.unknown(aval)
 * instead of PartialVal((None, pval)) we use PartialVal.known(pval)

Also disabled some new tests in random_tests.py on Mac. They segfault,
apparently due to the same issue #432.
2020-04-09 13:00:33 +03:00
Matthew Johnson
7f8ce8ff3c fix test errors from previous commit 2020-03-19 11:33:00 -07:00
George Necula
a955fd9dee Updated notebook that refered to freevars 2020-02-03 19:57:08 +01:00
George Necula
b0ffbaf1f6 Fixed also a notebook that has gone stale 2019-11-27 07:26:46 +01:00
Sharad Vikram
7bc2b0878a Update description of eqn.parmas 2019-11-08 13:11:17 -08:00
Sharad Vikram
1f40c9c4d2 Fix writing suggestions from mattjj 2019-11-03 15:54:05 -08:00
Sharad Vikram
e2e4e6e955 Fix title toc structure 2019-10-28 13:59:16 -07:00
Sharad Vikram
5d56999913 Add custom interpreter notebook 2019-10-28 13:58:55 -07:00