ROCm/jax

Pull Requests Commits

revert formating changes

JehandadKhan committed 1 year ago

0103489b

Deprecate obsolete gfx versions

JehandadKhan committed 1 year ago

44d5b782

add gfx1101 target

JehandadKhan committed 1 year ago

2e4d0ae6

[Mosaic GPU] Add lowering for `log`, and a fast path using log2.

Rifur13 committed 1 year ago

43b6be0e

Internal config change

kanglant committed 1 year ago

d112c85e

[JAX] Disable msan and asan for the profiler test running on nvidia gpu

Google-ML-Automation committed 1 year ago

8af6f70f

Merge pull request #26916 from jakevdp:update-array-api-tests

Google-ML-Automation committed 1 year ago

7d0aab5a

Merge pull request #26912 from dfm:resolve-args-error-message

Google-ML-Automation committed 1 year ago

4a73134b

Update array-api-tests to latest commit

jakevdp committed 1 year ago

f0bbd26d

Merge pull request #26765 from Qwlouse:patch-1

Google-ML-Automation committed 1 year ago

97db925a

Remove redundant `BUILD_TAG` from `JAX` wheels build rule.

Google-ML-Automation committed 1 year ago

ce3412e5

[Mosaic GPU] Make sure to do the async proxy fence before wargroup sync

apaszke committed 1 year ago

cdae5fcf

Improve error messages when input argument resolution fails in custom_* APIs.

dfm committed 1 year ago

8b1b039e

[pallas:triton] Emit a better error message for matmul with non-2D operands

superbobry committed 1 year ago

155839bb

Merge pull request #26883 from dfm:np-unique-sorted

Google-ML-Automation committed 1 year ago

8906f281

Update jnp.unique to support upstream interface changes.

dfm committed 1 year ago

6c5ef1a4

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

96dce0b6

Fix thread safety of JAX error checking

ayaka14732 committed 1 year ago

ea53c761

[Pallas/Fuser] Add support for custom_call_jvp/pjit to push_block_spec

sharadmv committed 1 year ago

00d9f452

Add fuser to jax.experimental.pallas

sharadmv committed 1 year ago

d32e282f

[Pallas] Add experimental (private for now) API for manual fusion into Pallas kernels

sharadmv committed 1 year ago

0b6c3550

Merge pull request #26865 from jakevdp:fix-indexing-error

Google-ML-Automation committed 1 year ago

2c7043f6

Merge pull request #26862 from jakevdp:logsumexp-docs

Google-ML-Automation committed 1 year ago

f9f47217

Merge pull request #26897 from jakevdp:cond-doc

Google-ML-Automation committed 1 year ago

4944dcb9

Merge pull request #26876 from carlosgmartin:fix_matrix_norm_empty_matrix

Google-ML-Automation committed 1 year ago

07d1cd02

Remove the skip for test_output_streaming_inside_scan

yashk2810 committed 1 year ago

07c4c03a

Merge pull request #26895 from hawkinsp:pad2

Google-ML-Automation committed 1 year ago

439c412c

doc: in lax.cond, note that both branches will be traced

jakevdp committed 1 year ago

84ca80d2

Fix wrong results in multidimensional pad.

hawkinsp committed 1 year ago

7f05b74b

Fix linalg.norm to return zero for proper norms of empty matrices.

carlosgmartin committed 1 year ago

897e1a13

Older