ROCm/jax

Pull Requests Commits

Require arraylike input for several jax.numpy functions

Jake VanderPlas committed 1 year ago

e95173a4

Merge pull request #21058 from jakevdp:jnp-delete-doc

a-googler committed 1 year ago

53208ffe

jnp.delete: better docs

jakevdp committed 1 year ago

88318e60

Merge pull request #21059 from jakevdp:numpy-doc-tests

a-googler committed 1 year ago

fc6a5d3a

Merge pull request #21064 from jakevdp:doc-new-tutorials

a-googler committed 1 year ago

20722248

DOC: replace old tutorials with new content

jakevdp committed 1 year ago

10ed827f

Refactor jax.numpy docstring tests

jakevdp committed 1 year ago

f2c2892c

Merge pull request #21057 from jakevdp:scipy-imports

a-googler committed 1 year ago

7e20e530

Remove last scipy imports

jakevdp committed 1 year ago

ff67e51e

Merge pull request #21055 from shoyer:relu-doc

a-googler committed 1 year ago

e70191bd

Recursively pull out __wrapped__ in linkcode_resolve()

shoyer committed 1 year ago

c77370c4

Merge pull request #21032 from mattjj:vmap-grad-shmap-bug

a-googler committed 1 year ago

c0cfc7ae

Merge pull request #21047 from shoyer:linkcode-robust

a-googler committed 1 year ago

989ea616

Merge pull request #21050 from rajasekharporeddy:test_branch3

a-googler committed 1 year ago

d05d29d8

Fix typos in docs and an error message

rajasekharporeddy committed 1 year ago

ccabdb29

Update XLA dependency to use revision

a-googler committed 1 year ago

52bf27d8

metal_plugin ci with jaxlib nightly

shuhand0 committed 1 year ago

28fe45d8

enable conv1d test

shuhand0 committed 1 year ago

99e5b8e9

Make linkcode_resolve() a bit more robust

shoyer committed 1 year ago

f6ee6127

update along with lax_numpy_test

shuhand0 committed 1 year ago

0c517f2d

Merge pull request #21042 from jakevdp:scipy-docs

a-googler committed 1 year ago

3952b7e7

Merge pull request #21043 from jakevdp:get-window

a-googler committed 1 year ago

653ee951

Merge pull request #21044 from mattjj:omnitracing-1-speedup

a-googler committed 1 year ago

ce25610f

[omnitracing] partially un-regress dispatch time

mattjj committed 1 year ago

e4c76c97

fix vmap-of-grad-of-shmap axis name reuse bug

mattjj committed 1 year ago

b7b26457

Remove remaining top-level scipy imports

jakevdp committed 1 year ago

27c0c41b

Made has_side_effect= parameter of mlir.emit_python_callback keyword-only

superbobry committed 1 year ago

03b733bd

Merge pull request #21013 from Micky774:array-api-trim

a-googler committed 1 year ago

187b2ac9

Merge pull request #21038 from rajasekharporeddy:test_branch1

a-googler committed 1 year ago

d5983e13

Merge pull request #21035 from olupton:A100-jet-test_dot

a-googler committed 1 year ago

db5cf1b9

Newer Older