ROCm/jax

Pull Requests Commits

Use HIPBLAS_V2 (#222)

Ruturaj4 committed 1 year ago

c16e4769

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

draganmladjenovic committed 1 year ago

Verified a8c11ba7

remove merge artifact leftovers (#267)

JehandadKhan committed 1 year ago

Verified 01c8d7fe

removing csv result compilation after Unit test... (#260)

zahiqbal committed 1 year ago

Verified 661e0e8f

add gfx1101 target (#252)

JehandadKhan committed 1 year ago

Verified 48f8e593

Fix RNN patch for 35-qa branch (#245)

Ruturaj4 committed 1 year ago

Verified fd1e518e

Fix bytes_limit key of mem_stats (#243)

Ruturaj4 committed 1 year ago

Verified c5ee54f9

Skip tests on ROCm (#235) (#242)

Ruturaj4 committed 1 year ago

Verified bbaaec08

Adding gpu_device_targets parameter (#233)

srayasam-amd committed 1 year ago

Verified f21c8260

Silence C23 warning, causing error in upb target for XLA (#230)

JehandadKhan committed 1 year ago

Verified a844a922

Change jaxlib/rocm/BUILD nanobind_extension to pybind (#226)

Ruturaj4 committed 1 year ago

Verified 39f7db68

[ROCm] Implement RNN support (#217)

Ruturaj4 committed 1 year ago

Verified dceb5310

Fix run_multi_gpu script multi-gpu issue and refactor code (#208)

Ruturaj4 committed 1 year ago

Verified da40f36d

Add gfx12xx archs (#205)

Ruturaj4 committed 1 year ago

Verified f99fbcb3

Merge pull request #201 from ROCm/rocm-jaxlib-v0.4.35-qa-fix2

kiran-thumma committed 1 year ago

Verified 5d56aaf1

buildkit option for docker build

kiran-thumma committed 1 year ago

Verified f075f904

Merge pull request #195 from ROCm/rocm-jaxlib-v0.4.35-qa-fix

kiran-thumma committed 1 year ago

Verified c5e32ac5

kiran-thumma committed 1 year ago

Verified 590ac67c

Merge pull request #169 from ROCm/ci_qa_sync

Ruturaj4 committed 1 year ago

Verified 8985f20e

[ROCm] Update XLA commit hash (#168)

Ruturaj4 committed 1 year ago

fab53b06

[ROCm] Remove exit on error (#167)

Ruturaj4 committed 1 year ago

20362f83

Add argument to override base docker in dockerfile

Ruturaj4 committed 1 year ago

010aea0c

[ROCm] ci build and dockerfile changes

Ruturaj4 committed 1 year ago

3b3083bb

[ROCm] Change run_multi_gpu set opts

Ruturaj4 committed 1 year ago

b8d2dede

BUILD file fixes and linter run

Ruturaj4 committed 1 year ago

a5a14175

[ROCm] fix jax and wheelhouse relative paths

Ruturaj4 committed 1 year ago

addca5d5

[ROCm] ci build fixes

Ruturaj4 committed 1 year ago

39c04506

JAX release 0.4.35

hawkinsp committed 1 year ago

81991d87

Use libtpu releases rather than libtpu-nightly for jax[tpu].

hawkinsp committed 1 year ago

e4f3f8f0

Merge pull request #24465 from jakevdp:fix-mypy

Google-ML-Automation committed 1 year ago

1c6b0a91

Older