ROCm/jax

Pull Requests Commits

revert formating changes

JehandadKhan committed 327 days ago

0103489b

Deprecate obsolete gfx versions

JehandadKhan committed 327 days ago

44d5b782

add gfx1101 target

JehandadKhan committed 333 days ago

2e4d0ae6

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

Rifur13 committed 334 days ago

43b6be0e

Internal config change

kanglant committed 334 days ago

d112c85e

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

Google-ML-Automation committed 334 days ago

8af6f70f

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

Google-ML-Automation committed 334 days ago

7d0aab5a

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

Google-ML-Automation committed 334 days ago

4a73134b

Update array-api-tests to latest commit

jakevdp committed 334 days ago

f0bbd26d

Merge pull request #26765 from Qwlouse:patch-1

Google-ML-Automation committed 334 days ago

97db925a

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

Google-ML-Automation committed 334 days ago

ce3412e5

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

apaszke committed 334 days ago

cdae5fcf

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

dfm committed 334 days ago

8b1b039e

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

superbobry committed 334 days ago

155839bb

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

Google-ML-Automation committed 334 days ago

8906f281

Update jnp.unique to support upstream interface changes.

dfm committed 334 days ago

6c5ef1a4

Update XLA dependency to use revision

Google-ML-Automation committed 334 days ago

96dce0b6

Fix thread safety of JAX error checking

ayaka14732 committed 335 days ago

ea53c761

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

sharadmv committed 335 days ago

00d9f452

Add fuser to jax.experimental.pallas

sharadmv committed 335 days ago

d32e282f

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

sharadmv committed 335 days ago

0b6c3550

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

Google-ML-Automation committed 335 days ago

2c7043f6

Merge pull request #26862 from jakevdp:logsumexp-docs

Google-ML-Automation committed 335 days ago

f9f47217

Merge pull request #26897 from jakevdp:cond-doc

Google-ML-Automation committed 335 days ago

4944dcb9

Merge pull request #26876 from carlosgmartin:fix_matrix_norm_empty_matrix

Google-ML-Automation committed 335 days ago

07d1cd02

Remove the skip for test_output_streaming_inside_scan

yashk2810 committed 335 days ago

07c4c03a

Merge pull request #26895 from hawkinsp:pad2

Google-ML-Automation committed 335 days ago

439c412c

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

jakevdp committed 335 days ago

84ca80d2

Fix wrong results in multidimensional pad.

hawkinsp committed 335 days ago

7f05b74b

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

carlosgmartin committed 335 days ago

897e1a13

Older