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

171 Commits

Author SHA1 Message Date
Roy Frostig
cd77debfa7 fix broken link in "NNs with TFDS" notebook 2022-07-28 19:04:08 -07:00
Ian McKenzie
0cc2ada432 Fix broken links for moved design_notes folder 2022-06-24 12:18:11 -07:00
Qiao Zhang
be71989af6 Remove broken image link. 
PiperOrigin-RevId: 456628284
 2022-06-22 15:54:36 -07:00
Jake VanderPlas
ed152953ef DOC: fix output repr in thinking_in_jax 2022-06-21 12:54:50 -07:00
fehiepsi
3ca0d3f149 Rename mesh into Mesh in xmap tutorial and doc 2022-06-03 17:11:09 +07:00
Jake VanderPlas
8a3bfe0300 DOC: add references for haskell-style signatures 2022-05-13 12:35:28 -07:00
Sharad Vikram
aca9dc6c6f Update custom interpreter tutorial 2022-05-11 11:11:47 -07:00
Jake VanderPlas
29efc30c63 DOC: update myst-nb to v0.15.0 2022-05-09 14:32:48 -07:00
Jake VanderPlas
dd71e2920b DOC: remove mentions of units 2022-05-04 16:08:44 -07:00
Jake VanderPlas
1134ff5871 CI: update jupytext to 0.13.8 2022-05-02 11:22:31 -07:00
Jake VanderPlas
72f14a5d64 Sharp bits: add miscellaneous divergences from numpy 2022-03-31 12:15:06 -07:00
Jake VanderPlas
32c718db11 quickstart: fix outdated TPU reference 2022-03-21 09:41:10 -07:00
Jake VanderPlas
74e7fdfd4c DOC: update sharp bits info about fori_loop differentiability 2022-02-25 09:15:55 -08:00
Jake VanderPlas
b9b79bab31 maint: update pre-commit package versions & fix new mypy errors 2022-01-31 13:39:11 -08:00
Matthew Johnson
726f60f6bc remove cpu platform setting in quickstart 
fixes #9244
 2022-01-19 10:24:09 -08:00
Mike McCoy
2e5ab11652 Resolves issue 8744 2022-01-12 21:10:45 +00:00
che-shr-cat
78977d6f5a fix broken links and update texts in thinking_in_jax.ipynb 2022-01-10 16:19:57 +03:00
Jake VanderPlas
4a72e57ce0 DOC: add FAQ section on JAX vs. Numpy performance 2021-11-24 12:04:02 -08:00
Peter Hawkins
58199b4b9a Delete the XLA in Python notebook. 
Its tests are failing, and it describes a non-public API that we are phasing out.
 2021-11-18 09:45:06 -05:00
Peter Hawkins
e783cbcb72 Port remaining translation rules inside JAX to new style. 
PiperOrigin-RevId: 404288551
 2021-10-19 09:48:37 -07:00
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