mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
23,091 Commits 503 Branches 263 Tags
Commit Graph

64 Commits

Author SHA1 Message Date
Michael Hudgins
d4d1518c3d Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax 
PiperOrigin-RevId: 676843138
 2024-09-20 07:52:33 -07:00
Roy Frostig
763d600508 docs: remove inline authors/dates on misc doc pages 
Git history covers this better and automatically.
 2024-08-28 11:04:14 -07:00
Jake VanderPlas
68be5b5085 CI: update ruff to v0.6.1 2024-08-27 14:54:11 -07:00
Roy Frostig
09e73118bf docs: more sentence case 2024-08-12 20:07:49 -07:00
Dan Foreman-Mackey
705eed3388 Fixing dtype canonicalization in sharp edges tutorial. 
As reported in https://github.com/google/jax/issues/22493, the sharp
edges tutorial doesn't seem to actually enable x64 when it says it does.

Fixes https://github.com/google/jax/issues/22493
 2024-07-22 15:02:02 -04:00
jax authors
fc1e1d4a65 Add freshness metablock to JAX OSS docs. 
PiperOrigin-RevId: 645508135
 2024-06-21 14:50:49 -07:00
George Necula
0223e830c1 [docs] Enable doctest for .md documentation files 
The current invocation of doctest from GH actions picked up only .rst files.
We enable .md files also, and we make a few changes to ensure that the
doctest passes on the existing files.

The changes fall into several categories:
  * add a newline before the end of the code block, for doctest to
    pick up the expected output properly
  * update the expected values to match the current behavior
  * disable some doctests that raise expected exceptions, whenever
    I could not get doctest to match the exception details.
    Sometimes +IGNORE_EXCEPTION_DETAIL was enough, and other times
    I had to use +SKIP.
 2024-06-03 11:27:34 +03:00
Jake VanderPlas
f090074d86 Avoid 'from jax import config' imports 
In some environments this appears to import the config module rather than
the config object.
 2024-04-11 13:23:27 -07:00
rajasekharporeddy
d92e735001 Fix Typos in docs 2024-04-11 07:54:13 +05:30
Roy Frostig
75a53f46e0 update notebooks to new-style typed PRNG keys 2024-03-07 12:40:10 -08:00
jax authors
2cfbd35149 Typo: MT19937 is 2.5 kilobytes, not kilobits. 
PiperOrigin-RevId: 603903613
 2024-02-03 02:03:00 -08:00
Peter Hawkins
2c32660a8f Replace references to DeviceArray with Array. 
A number of stale references are lurking in our documentation.
 2023-08-18 17:46:00 -04:00
Jake VanderPlas
fbe4f10403 Change to simpler import for jax.config 2023-04-21 11:51:22 -07:00
Jake VanderPlas
c7c9cb652e Sharp bits: refer to ndarray.at in out-of-bound indexing discussion 2023-03-17 13:29:05 -07:00
Peter Hawkins
71f120beed Add "Open in Kaggle" buttons to Jupyter notebooks. 2023-03-01 13:15:42 -05:00
Jake VanderPlas
c4ec2996af Sharp bits: mention alternatives to lax.cond 2023-02-02 13:19:26 -08:00
Jake VanderPlas
9e355a6606 Sharp Bits: add section on Dynamic shapes 2023-01-19 11:37:03 -08:00
Jake VanderPlas
b2141229e3 Sharp bits: raise exceptions directly 2023-01-17 14:13:38 -08:00
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
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
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
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
Jake VanderPlas
b04d0c75a9 sync common gotchas notebook 2021-05-21 12:24:28 -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
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
Justin Lebar
e8c7d9e281
s/Three-fry/Threefry/ (#3918) 
Per http://www.thesalmons.org/john/random123/
 2020-07-30 16:52:44 -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
Jake Vanderplas
41292a2af6
mention numpy & scipy convolve functions in gotchas doc. (#3214) 2020-05-27 13:20:27 -07:00
Jascha Sohl-Dickstein
190f88dede
Update Common_Gotchas_in_JAX.ipynb (#3189) 
typo fix
 2020-05-22 14:12:44 -07:00
yurodiviy
56f6294e37
Implement nanargmin-max and add tests (#2398) 
Co-authored-by: vlad <veryfakemail@ya.ru>
 2020-04-28 15:23:03 -04:00