mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 11:36:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
7,285 Commits 508 Branches 263 Tags
Commit Graph

121 Commits

Author SHA1 Message Date
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
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
frederikwilde
42cbe49ce6
Correction a typo of the period of the PRNG. (#3578) 2020-06-26 15:19:57 -07:00
8bitmp3
7f9fc27c48
Another small fix of the link rendering in the Autodiff Cookbook - vmap transformation (#3502) 2020-06-21 13:52:34 -07:00
8bitmp3
f0dff9c19b
Fix a link rendering to Autograd's reverse-mode Jacobian method (#3501) 2020-06-21 11:07:29 -07:00
Matthew Johnson
0c29cc15b9 fix typos 2020-06-14 10:13:56 -07:00
Matthew Johnson
1b88fba57c fix and better explain complex JVPs / VJPs 
fixes #3433
 2020-06-14 10:09:15 -07:00
Matthew Johnson
d3ccf0a5c7 fix typo in docs 2020-06-09 07:35:07 -07:00
Matthew Johnson
2a6d3f417c fix another docs bug 2020-06-08 14:13:01 -07:00
Matthew Johnson
0a716978ad fix a docs bug 2020-06-08 13:30:00 -07:00
Adam Paszke
7a0e8bae30 Update notebooks 2020-06-05 15:52:03 +00:00