mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 12:46:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
9,381 Commits 508 Branches 263 Tags
Commit Graph

151 Commits

Author SHA1 Message Date
Basil
6fac5d36a1 Remove unused variable 2021-10-13 14:38:33 +01:00
Peter Hawkins
104a46594b Add DeprecationWarnings to jax.ops.index_... operators. 
Remove uses of index_... in Common Gotchas notebook.
 2021-10-05 20:47:22 -04:00
Jake VanderPlas
198cc5ee4f chore: update jupytext to v0.1.13 & re-sync notebooks 2021-10-05 14:30:16 -07:00
Peter Hawkins
2c2f4033cc Move contents of jax.lib to jax._src.lib. 
Add shim libraries for functions exported from jax.lib that other code seems to use in practice.

PiperOrigin-RevId: 398471863
 2021-09-23 06:33:55 -07:00
jax authors
394ef618d5 Merge pull request #7868 from PWhiddy:main 
PiperOrigin-RevId: 397197685
 2021-09-16 16:43:35 -07:00
PWhiddy
af22aae1e0 fix html formatting of markdown 2021-09-15 16:13:02 -04:00
Qiao Zhang
aec40b48d3 Update docs for nan checker and tfds example. 2021-09-15 11:54:48 -07:00
Roy Frostig
e659c54977 docs: remove application tutorials 2021-09-10 18:52:47 -07:00
Jake VanderPlas
9959d087c1 maml.ipynb: fix sync issue 2021-08-24 09:11:40 -07:00
Xiaohu Zhu
db0efd917f
Update maml.md 
typo fixed.
 2021-08-21 15:50:36 +08:00
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07:00
Shawn Presser
7b879c811d Fix typo in xmap_tutorial 2021-06-29 17:09:57 -05: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
jax authors
1912bd97f4 Merge pull request #6673 from hypercubestart:master 
PiperOrigin-RevId: 377417011
 2021-06-03 18:29:31 -07:00
Peter Hawkins
46cc654537 Move jax.abstract_arrays to jax._src.abstract_arrays. 
PiperOrigin-RevId: 377044255
 2021-06-02 06:25:22 -07:00
Phúc Lê Khắc
f30a36dbbc Typo. 
Update Common_Gotchas_in_JAX.md
 2021-05-29 17:22:40 +01:00
Jake VanderPlas
d844609c6d DOC: add section to Sharp Bits discussing implicit list conversions 2021-05-26 09:03:42 -07:00
Jake VanderPlas
b04d0c75a9 sync common gotchas notebook 2021-05-21 12:24:28 -07:00
James Bradbury
8c0aa3880b add note on autodiff of OOB indexing 2021-05-21 12:23:33 -07:00
Jake VanderPlas
b624731203 Fix broken link in custom derivatives notebook 2021-05-21 11:32:21 -07:00
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
Andrew Liu
8d54a53b53 Update convolutions.md 
Update convolutions.ipynb

Update convolutions.ipynb

Update convolutions.md
 2021-05-11 17:00:09 -07: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