ROCm/jax

Pull Requests Commits

Fix C++23 build errors (#257) (#318)

mrodden committed 265 days ago

85e80e42

Fix permissions on /tmp (#320)

gabeweisz committed 265 days ago

7aa324ce

Update offload ROCm architectures (#311)

Ruturaj4 committed 265 days ago

ca4cfe37

Set higher tolerance for failing fp8 tests (#306)

Ruturaj4 committed 265 days ago

defaf1a5

Fixes for 0.5.0 build (#308)

mrodden committed 265 days ago

52912e63

Add support for ROCm wheel based install (#283)

mrodden committed 265 days ago

52f0201e

Fix and enable RNN tests on JAX 0.5.0 branch (#301)

Ruturaj4 committed 265 days ago

367c08c1

Fix the latest XLA has for 0.5.0 JAX branch (#302)

Ruturaj4 committed 265 days ago

6dee2704

disable c23-extension warning so clang-19 works

gabeweisz committed 274 days ago

3d4f91d0

Fix dev build script (#286)

gulsumgudukbay committed 278 days ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 279 days ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 282 days ago

Verified a496792a

[pallas:triton] Fix atomic min/max lowering for unsigned integers and float types (#264)

draganmladjenovic committed 286 days ago

Verified 8445686b

removing pytest-csv and removing csv logs from unit testing (#261)

zahiqbal committed 290 days ago

Verified 9ce2aec6

add gfx1101 target (#249) (#254)

JehandadKhan committed 290 days ago

Verified 48a36151

Fix sha256 hash to point to the latest xla commit (#251)

Ruturaj4 committed 293 days ago

Verified 772bd4ed

Skip tests on ROCm (#235)

Ruturaj4 committed 304 days ago

Verified 790fe78b

Fix invalid lowerings for ROCm in Pallas (#223)

mrodden committed 310 days ago

Verified 6b351552

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 318 days ago

Verified 1fc3d157

Merge pull request #215 from ROCm/5.0_xla_sha_fix

zahiqbal committed 327 days ago

Verified b6843ff8

Update workspace.bzl

Ruturaj4 committed 327 days ago

Verified 1c9f9073

Update XLA hash for 0.5.0 branch (#212)

Ruturaj4 committed 331 days ago

Verified cdf1ef5a

Merge pull request #211 from ROCm/jax_5.0_xla_sha_fix

zahiqbal committed 332 days ago

Verified 81ac86ba

Update XLA hash for 0.5.0 branch

Ruturaj4 committed 332 days ago

b51b0c76

Release JAX 0.5.0

hawkinsp committed 338 days ago

c25fb92c

Reverts f1b894d14a28ac22a037fb79177b991275c75a18

Google-ML-Automation committed 338 days ago

a527aba6

[sharding_in_types] Error out for reshape for splits like this: `(4, 6, 8)` -> `(4, 4, 2, 6)`

yashk2810 committed 338 days ago

ce85b898

Update XLA dependency to use revision

Google-ML-Automation committed 338 days ago

7cac76d3

[Mosaic GPU] Delete unused declarations of `mosaic_gpu_memcpy_async_h2d`.

bchetioui committed 338 days ago

d3be190e

[mosaic_gpu] Added a serialization pass

superbobry committed 338 days ago

d34c40f6

Older