ROCm/jax

Pull Requests Commits

Add canonicalization of python versions

mrodden committed 266 days ago

8c1eb0a2

Split parse_gpu_targets test cases

mrodden committed 266 days ago

3a2fec78

Fix missing defaults for GPU target arch'es (#293)

mrodden committed 266 days ago

d0f4d319

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

mrodden committed 268 days ago

Verified 0074fed2

Fix permissions on /tmp (#320)

gabeweisz committed 268 days ago

Verified 82a2ec71

Update offload ROCm architectures (#311)

Ruturaj4 committed 269 days ago

Verified 7613a24b

Set higher tolerance for failing fp8 tests (#306)

Ruturaj4 committed 270 days ago

Verified 6a539471

Fixes for 0.5.0 build (#308)

mrodden committed 273 days ago

Verified 887d1ede

Add support for ROCm wheel based install (#283)

mrodden committed 273 days ago

Verified 0218c7bd

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

Ruturaj4 committed 274 days ago

Verified 41bcf9f1

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

Ruturaj4 committed 274 days ago

Verified ce337284

Fix dev build script (#286)

gulsumgudukbay committed 277 days ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 277 days ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 281 days ago

Verified a496792a

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

draganmladjenovic committed 284 days ago

Verified 8445686b

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

zahiqbal committed 289 days ago

Verified 9ce2aec6

add gfx1101 target (#249) (#254)

JehandadKhan committed 289 days ago

Verified 48a36151

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

Ruturaj4 committed 291 days ago

Verified 772bd4ed

Skip tests on ROCm (#235)

Ruturaj4 committed 302 days ago

Verified 790fe78b

Fix invalid lowerings for ROCm in Pallas (#223)

mrodden committed 308 days ago

Verified 6b351552

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 317 days ago

Verified 1fc3d157

Merge pull request #215 from ROCm/5.0_xla_sha_fix

zahiqbal committed 326 days ago

Verified b6843ff8

Update workspace.bzl

Ruturaj4 committed 326 days ago

Verified 1c9f9073

Update XLA hash for 0.5.0 branch (#212)

Ruturaj4 committed 330 days ago

Verified cdf1ef5a

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

zahiqbal committed 331 days ago

Verified 81ac86ba

Update XLA hash for 0.5.0 branch

Ruturaj4 committed 331 days ago

b51b0c76

Release JAX 0.5.0

hawkinsp committed 336 days ago

c25fb92c

Reverts f1b894d14a28ac22a037fb79177b991275c75a18

Google-ML-Automation committed 336 days ago

a527aba6

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

yashk2810 committed 336 days ago

ce85b898

Update XLA dependency to use revision

Google-ML-Automation committed 336 days ago

7cac76d3

Older