ROCm/jax

Pull Requests Commits

Add support for ROCm wheel based install

mrodden committed 291 days ago

8afc41c1

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

mrodden committed 295 days ago

Verified ef1d5618

Fix gpu targets arg issue (#291)

mrodden committed 295 days ago

Verified 792fceea

Fix auditwheel version issue (#289)

mrodden committed 295 days ago

Verified 993adb49

Update commit hash (#287)

Ruturaj4 committed 295 days ago

Verified 0c941580

35 several uts (#282)

Ruturaj4 committed 298 days ago

Verified 8c48c0f6

Fix extend test for ROCm (#279)

Ruturaj4 committed 299 days ago

Verified 9e785a07

Temporarily allow deprecation warnings for `scipy.special.lpmn` and `scipy.special.sph_harm`. (#278)

Ruturaj4 committed 299 days ago

Verified b5c700c9

Use HIPBLAS_V2 (#222) (#277)

Ruturaj4 committed 300 days ago

Verified d90651ba

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

draganmladjenovic committed 302 days ago

Verified a8c11ba7

remove merge artifact leftovers (#267)

JehandadKhan committed 305 days ago

Verified 01c8d7fe

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

zahiqbal committed 307 days ago

Verified 661e0e8f

add gfx1101 target (#252)

JehandadKhan committed 307 days ago

Verified 48f8e593

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

Ruturaj4 committed 313 days ago

Verified fd1e518e

Fix bytes_limit key of mem_stats (#243)

Ruturaj4 committed 314 days ago

Verified c5ee54f9

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

Ruturaj4 committed 314 days ago

Verified bbaaec08

Adding gpu_device_targets parameter (#233)

srayasam-amd committed 320 days ago

Verified f21c8260

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

JehandadKhan committed 327 days ago

Verified a844a922

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

Ruturaj4 committed 333 days ago

Verified 39f7db68

[ROCm] Implement RNN support (#217)

Ruturaj4 committed 333 days ago

Verified dceb5310

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

Ruturaj4 committed 352 days ago

Verified da40f36d

Add gfx12xx archs (#205)

Ruturaj4 committed 357 days ago

Verified f99fbcb3

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

kiran-thumma committed 358 days ago

Verified 5d56aaf1

buildkit option for docker build

kiran-thumma committed 360 days ago

Verified f075f904

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

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

Older