ROCm/jax

Pull Requests Commits

Fixes for 0.5.0 build

mrodden committed 315 days ago

c23a8146

Add support for ROCm wheel based install (#283)

mrodden committed 315 days ago

Verified 0218c7bd

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

Ruturaj4 committed 316 days ago

Verified 41bcf9f1

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

Ruturaj4 committed 316 days ago

Verified ce337284

Fix dev build script (#286)

gulsumgudukbay committed 319 days ago

Verified d9ec58fd

Fix auditwheel version issue (#290)

mrodden committed 319 days ago

Verified 1c96ac20

Deprecate unused pre-production gfx versions (#272)

JehandadKhan committed 323 days ago

Verified a496792a

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

draganmladjenovic committed 326 days ago

Verified 8445686b

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

zahiqbal committed 331 days ago

Verified 9ce2aec6

add gfx1101 target (#249) (#254)

JehandadKhan committed 331 days ago

Verified 48a36151

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

Ruturaj4 committed 333 days ago

Verified 772bd4ed

Skip tests on ROCm (#235)

Ruturaj4 committed 344 days ago

Verified 790fe78b

Fix invalid lowerings for ROCm in Pallas (#223)

mrodden committed 350 days ago

Verified 6b351552

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 359 days ago

Verified 1fc3d157

Merge pull request #215 from ROCm/5.0_xla_sha_fix

zahiqbal committed 1 year ago

Verified b6843ff8

Update workspace.bzl

Ruturaj4 committed 1 year ago

Verified 1c9f9073

Update XLA hash for 0.5.0 branch (#212)

Ruturaj4 committed 1 year ago

Verified cdf1ef5a

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

zahiqbal committed 1 year ago

Verified 81ac86ba

Update XLA hash for 0.5.0 branch

Ruturaj4 committed 1 year ago

b51b0c76

Release JAX 0.5.0

hawkinsp committed 1 year ago

c25fb92c

Reverts f1b894d14a28ac22a037fb79177b991275c75a18

Google-ML-Automation committed 1 year ago

a527aba6

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

yashk2810 committed 1 year ago

ce85b898

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

7cac76d3

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

bchetioui committed 1 year ago

d3be190e

[mosaic_gpu] Added a serialization pass

superbobry committed 1 year ago

d34c40f6

[sharding_in_types] Rename `.at[...].get(out_spec)` to `.at[...].get(out_sharding)`.

yashk2810 committed 1 year ago

af667199

Rename out_type -> out_sharding parameter on einsum

yashk2810 committed 1 year ago

97cd7483

Replace Auto/User/Collective AxisTypes names with Hidden/Visible/Collective.

yashk2810 committed 1 year ago

49224d6c

[Mosaic TPU] Use large to compact 2nd minor retiling for conversions going both ways

apaszke committed 1 year ago

bd22bfef

Add quotes to pip commands in docs around option install for zsh

rdyro committed 1 year ago

aa9cea0a

Older