Jake VanderPlas
ff7c2afcaf
DOC: directly silence GPU warning
2021-05-21 10:03:50 -07:00
Bogdan Damoc
db39a67cca
Update documentation for custom_jvp handling of nondiff_argnums as arguments of _fwd and _bwd rules.
2021-05-12 16:44:52 +01:00
jax authors
cb881bb4f0
Merge pull request #6425 from apaszke:notebook-fix
...
PiperOrigin-RevId: 368451960
2021-04-14 10:00:28 -07:00
Adam Paszke
62ac3e058b
Fix inline image embedding in the xmap tutorial
2021-04-14 10:14:50 +00:00
Peter Hawkins
26e9ebcdae
Move jax.api to jax._src.api.
...
PiperOrigin-RevId: 368233837
2021-04-13 09:43:24 -07:00
Sharad Vikram
577c601191
Update references in custom interpreter notebook
2021-04-12 19:00:28 -07:00
Dan Zheng
0b41ff93ea
DOC: fix typos.
2021-04-09 23:53:55 -04:00
jax authors
a27b9005c5
Merge pull request #6100 from apaszke:xmap-tutorial
...
PiperOrigin-RevId: 367613958
2021-04-09 05:10:12 -07:00
Adam Paszke
40253f471d
Make xmap tutorial public
...
Note that xmap is still experimental, so treat this more as a preview
than as documentation.
2021-04-08 15:56:58 +00:00
Niklas Schmitz
4b5265b4d1
fix markdown math formatting
2021-03-18 10:02:20 +01:00
Jake VanderPlas
4d8e9540a4
Sharp Bits: fix formatting of code blocks within list
2021-03-16 14:21:21 -07:00
Jake VanderPlas
39fb1b7f00
DOC: fix deprecated cond() signature in Common_Gotchas notebook
2021-03-04 13:04:30 -08:00
Jake VanderPlas
8f5038d4b9
DOC: add initial JAX glossary
2021-03-01 17:56:43 -08:00
Jake VanderPlas
24e920e3f5
change colab.sandbox->colab.research
2021-02-25 10:29:43 -08:00
Jake VanderPlas
067be89a0c
DOC: minor documentation & formatting fixes
2021-02-23 10:31:44 -08:00
jax authors
2bf97298be
Merge pull request #5765 from jakevdp:cleanup-notebooks
...
PiperOrigin-RevId: 358847147
2021-02-22 10:21:42 -08:00
jax authors
25c03f79f2
Merge pull request #5807 from jakevdp:fix-xla-notebook
...
PiperOrigin-RevId: 358844761
2021-02-22 10:11:47 -08:00
Jake VanderPlas
afaec32969
fix conflicts
2021-02-22 10:08:33 -08:00
Jake VanderPlas
586b3bb6a9
DOC: regularize notebook metadata
2021-02-22 10:04:42 -08:00
Jake VanderPlas
d9af9ee487
Fix stray header in XLA_In_Python notebook
2021-02-22 09:05:48 -08:00
James Reed
c19486c900
Fix typo in thinking_in_jax notebook
2021-02-20 13:24:26 -08:00
Noah Fiedel
bb5ae556ad
Update thinking_in_jax.md
2021-02-19 21:59:47 -08:00
Noah Fiedel
80a8204fdb
Update thinking_in_jax.ipynb
...
Fix typo in notebook url.
2021-02-19 12:37:04 -08:00
Jake VanderPlas
236ba14585
move convolutions from common gotchas to its own file
2021-02-16 17:21:56 -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
Jake VanderPlas
7545aeb8d3
doc build: avoid executing thinking_in_jax.ipynb
2021-02-04 09:26:11 -08:00
Jake VanderPlas
667cf9bcb1
Merge problematic cells in thinking_in_jax.ipynb
2021-02-03 09:25:26 -08:00
Jake VanderPlas
113584c8b0
Add hidden sleep statement in attempt to sidestep RTD build issue
2021-02-01 14:44:18 -08:00
Jake VanderPlas
967f3ac7a3
Add thinking_in_jax.ipynb
2021-01-26 12:08:37 -08:00
Peter Hawkins
3ac809ede3
[JAX] Move jax.util to jax._src_util.
...
PiperOrigin-RevId: 351234602
2021-01-11 14:21:07 -08:00
Qiao Zhang
533b4ac3c4
Updates XLA_in_Python notebook with the latest API.
2021-01-07 10:14:50 -08:00
jax authors
4e0b8a8a10
Merge pull request #4948 from gmittal:master
...
PiperOrigin-RevId: 346291917
2020-12-08 04:01:02 -08:00
Gautam Mittal
df411ff49b
Update autodiff_cookbook.ipynb
2020-12-07 10:06:08 -08:00
Gautam Mittal
4440ef4568
Move example
2020-12-07 10:05:00 -08:00
jax authors
3a4e35f4e0
Internal change
...
PiperOrigin-RevId: 346099032
2020-12-07 08:31:40 -08:00
Gautam Mittal
b2a7be9806
Update quickstart.ipynb
2020-11-18 16:22:07 -08:00
Gautam Mittal
f54e531cc4
Add vector-valued gradient example.
2020-11-18 16:18:43 -08:00
Matthew Johnson
491ad9657d
fix typo in autodiff cookbook
2020-11-12 14:33:17 -08:00
Matthew Johnson
e51163af32
only pass hashable values as static args
2020-10-16 13:11:56 -07:00
jax authors
4a20eea828
Copybara import of the project:
...
--
609f6f3e16d21fed34cc5269c54a0d78ac44a8bc by Matthew Johnson <mattjj@google.com>:
fix custom_jvp/vjp closure issues
PiperOrigin-RevId: 337457689
2020-10-16 00:21:32 -07:00
johnpjf
b357005e60
Update Common_Gotchas_in_JAX.ipynb
...
Clarify that the index is clamped to the bounds of the array when accessing out of bounds.
2020-10-05 10:19:18 -07:00
Qiao Zhang
26a53ae554
Add comments for residuals from f_bwd. ( #4244 )
2020-09-10 13:58:28 +03:00
George Thomas
b4efb31f1a
Docs: Fix broken link in quickstart ( #4102 )
2020-08-19 11:36:28 -07:00
Matthew Johnson
09d8ac14de
use fewer internal APIs in custom interp notebook ( #4016 )
2020-08-10 20:48:03 -07:00
Justin Lebar
e8c7d9e281
s/Three-fry/Threefry/ ( #3918 )
...
Per http://www.thesalmons.org/john/random123/
2020-07-30 16:52:44 -07:00
Stephan Hoyer
dd7ab39e4d
Fix formatting in the custom derivatives notebook ( #3876 )
...
Sphinx is apparently quite picky about consistent use of headers: you can't
skip a header level. We were getting warnings like "WARNING: Title level
inconsistent" in the docs build, and sub-headers weren't showing up on this
page after the first section.
2020-07-27 22:25:16 -07:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks ( #3760 )
2020-07-15 13:17:38 -07:00
igorwilbert
c485a5b04a
Improve jax_debug_nans documentation ( #3665 )
2020-07-06 09:23:01 +03:00
igorwilbert
3d59a0bcba
Adds information about iterator in fori_loop in The Sharp Bits documentation ( #3632 )
2020-07-04 15:57:15 +03:00
