Matthew Johnson
f5723848d3
fix error in autodiff cookbook: 3x not 2x
2019-12-30 07:36:36 -08:00
David Bieber
30bede1f6a
fix typo in autodiff cookbook ( #1921 )
2019-12-27 11:02:06 -08:00
Matthew Johnson
8dad859e04
streamline readme, add pmap
2019-12-14 10:32:04 -08:00
tamaranorman
26e863923a
Support atrous conv in same padded convolution and add warning if use transposed convolution with same or valid padding. ( #1806 )
...
PiperOrigin-RevId: 283517237
2019-12-09 08:06:59 -08:00
George Necula
b0ffbaf1f6
Fixed also a notebook that has gone stale
2019-11-27 07:26:46 +01:00
George Necula
8777864c96
Minor edits
2019-11-24 20:29:44 +01:00
George Necula
b12a8019c8
Update docs/notebooks/JAX_pytrees.ipynb
...
Co-Authored-By: Stephan Hoyer <shoyer@google.com>
2019-11-24 20:29:29 +01:00
George Necula
4e89d43a75
Added JAX pytrees notebook
...
Also added docstrings to the tree_util module.
2019-11-24 20:29:07 +01:00
Peter Buchlovsky
9d1204689f
Fix typo
2019-11-20 08:53:01 -08:00
Peter Buchlovsky
410ebfeb1c
Fix typo
2019-11-20 08:52:46 -08:00
Anselm Levskaya
f882359511
fix lax.scan notes in gotchas notebook
...
Note that lax.scan is now jittable and differentiable in the Gotchas notebook.
2019-11-17 00:19:24 -08:00
Matthew Johnson
7a9ea8a006
Merge pull request #1582 from sharadmv/custom-interpreter
...
Add custom interpreter notebook
2019-11-08 13:13:24 -08: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
George Necula
8880e262b0
Use redthedocs links for Colabs ( #1572 )
...
Steer the documentation readers to readthedocs.
Also, minor fixes to the wording of How_jax_primitives_work, suggested by Dougal
2019-10-29 08:53:35 +01: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
George Necula
eae59d0b2c
Moved all notebooks to docs/notebooks. ( #1493 )
...
* Moved all notebooks to docs/notebooks.
Now all notebooks are in the same place, thus all are subject
to auto-doc generation at readthedocs.io and to automated testing
with travis.
Some notebooks are too slow, exclude them at docs/conf.py:exclude_patterns.
Cleanup a bit the section headings in notebooks so that they show
up well in readtehdocs.io.
* Increase the cell timeout for executing notebooks
* Exclude also the neural network notebook from auto-generation (timing out)
* Disable the score_matching notebook from auto-doc (travis does not have sklearn)
2019-10-17 08:58:25 +02:00
Trevor Cai
c50495b272
Fix List rendering in RTD for primitives.ipynb ( #1501 )
...
Colab doesn't require a newline before unordered list in Markdown; RTD
does.
2019-10-14 10:55:13 -07:00
George Necula
a5063ee68b
Merge pull request #1423 from gnecula/colabs
...
Added How_JAX_primitives_work colab
2019-10-07 14:59:37 +02:00
George Necula
c4e4199508
Fix typo in notebook JSON ( #1441 )
...
Perhaps this is the reason why readthedocs does not show the link to the notebook?
2019-10-04 08:38:36 -07:00
George Necula
73144fafb5
Update docs/notebooks/How_JAX_primitives_work.ipynb
...
Co-Authored-By: James Bradbury <jekbradbury@google.com>
2019-10-04 11:58:07 +02:00
George Necula
ef886c7fcc
Update docs/notebooks/How_JAX_primitives_work.ipynb
...
Co-Authored-By: James Bradbury <jekbradbury@google.com>
2019-10-04 11:56:06 +02:00
Roy Frostig
db3b0dd036
fix missing delimiter in notebook source
2019-10-03 14:02:45 -07:00
George Necula
0fdf377343
Switched testing to using sphinx (part of the documentation generation)
2019-10-03 20:54:28 +02:00
George Necula
57e39ce938
Added support for testing Colab notebooks.
...
Testing is done by running "jupyter nbconvert --to notebook" and
then parsing the resulting notebook to look for errors.
One can declare expected errors, and the test will fail if those
are missing.
In the process of doig this, found and fixed a bug in the autodiff_cookbook
notebook.
2019-10-03 20:07:49 +02:00
George Necula
7bc0f70f87
Fix the title to show nicely in sphinx
2019-10-03 11:59:51 +02:00
George Necula
75c2236063
Addressed comments for the Colab.
...
* Cleaned up use of section levels
* Renamed ma to multiply_add and sq_add to square_add
* Other minor clarifications
* Separated the Colabs into Tutorials and Advanced Tutorials
2019-10-03 11:20:04 +02:00
George Necula
454320e9c9
Added How_JAX_primitives_work colab
2019-10-02 14:42:01 +02:00
Stephan Hoyer
9bd7330e1f
Notebooks on RTD ( #1121 )
2019-09-30 11:00:02 -07:00
