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

26 Commits

Author SHA1 Message Date
Peter Hawkins
71b29b1cc6 Create JAX Enhancement Proposals (JEPs). 
Migrate existing design documents to JEPs.
 2022-08-08 16:13:58 -04:00
Matthew Johnson
7ed4fa550e in sharp bits notebook, add back iaml blog post link 2022-08-04 08:29:26 -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
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
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
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
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
elliotwaite
7392a57b75 DOC: many small fixes 2021-08-04 16:55:13 -07: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
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
James Bradbury
8c0aa3880b add note on autodiff of OOB indexing 2021-05-21 12:23:33 -07: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
24e920e3f5 change colab.sandbox->colab.research 2021-02-25 10:29:43 -08:00
Jake VanderPlas
586b3bb6a9 DOC: regularize notebook metadata 2021-02-22 10:04:42 -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