ROCm/jax

Pull Requests Commits

update XLA commit hash with gfx1101 changes

JehandadKhan committed 307 days ago

0e316755

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

zahiqbal committed 309 days ago

Verified 661e0e8f

add gfx1101 target (#252)

JehandadKhan committed 309 days ago

Verified 48f8e593

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

Ruturaj4 committed 315 days ago

Verified fd1e518e

Fix bytes_limit key of mem_stats (#243)

Ruturaj4 committed 316 days ago

Verified c5ee54f9

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

Ruturaj4 committed 316 days ago

Verified bbaaec08

Adding gpu_device_targets parameter (#233)

srayasam-amd committed 323 days ago

Verified f21c8260

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

JehandadKhan committed 329 days ago

Verified a844a922

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

Ruturaj4 committed 335 days ago

Verified 39f7db68

[ROCm] Implement RNN support (#217)

Ruturaj4 committed 335 days ago

Verified dceb5310

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

Ruturaj4 committed 354 days ago

Verified da40f36d

Add gfx12xx archs (#205)

Ruturaj4 committed 359 days ago

Verified f99fbcb3

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

kiran-thumma committed 360 days ago

Verified 5d56aaf1

buildkit option for docker build

kiran-thumma committed 362 days 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

Merge pull request #21524 from andportnoy:aportnoy/unknown-platform-lowering-warning

Google-ML-Automation committed 1 year ago

9a2dd19a

Merge pull request #24452 from jakevdp:insert-doc

Google-ML-Automation committed 1 year ago

1e41d5ef

Older