ROCm/jax

Pull Requests Commits

Update ci_build

srayasam-amd committed 310 days ago

Verified 35ffecdc

Update build_wheels.py

srayasam-amd committed 310 days ago

Verified efa7bf4f

Update ci_build

srayasam-amd committed 310 days ago

Verified 53c845d3

Update ci_build.sh

srayasam-amd committed 310 days ago

Verified aba4376e

Update ci_build

srayasam-amd committed 312 days ago

Verified 40a99379

Update ci_build

srayasam-amd committed 312 days ago

Verified ae9f449c

Update build_wheels.py

srayasam-amd committed 312 days ago

Verified c95d2676

Update ci_build

srayasam-amd committed 312 days ago

Verified 90f95a2e

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 312 days ago

Verified 2aa8e23b

Update ci_build.sh

srayasam-amd committed 312 days ago

Verified 96315972

Update ci_build.sh

srayasam-amd committed 312 days ago

Verified 082d0575

Update ci_build.sh

srayasam-amd committed 312 days ago

Verified d3cdd4f7

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 312 days ago

Verified 6f8b984b

Update ci_build.sh

srayasam-amd committed 312 days ago

Verified 3e64d0b6

Update ci_build.sh

srayasam-amd committed 312 days ago

Verified 6d50f2a2

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 312 days ago

Verified bdea57c4

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 312 days ago

Verified d917a225

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 313 days ago

Verified 3ba06823

Update ci_build.sh

srayasam-amd committed 313 days ago

Verified 58f2c0e4

Update ci_build.sh

srayasam-amd committed 313 days ago

Verified 3403473e

Update ci_build.sh

srayasam-amd committed 313 days ago

Verified 50013656

Update ci_build.sh

srayasam-amd committed 313 days ago

Verified a53822fa

Update ci_build

srayasam-amd committed 315 days ago

Verified 5da1c4e8

Update build_wheels.py

srayasam-amd committed 315 days ago

Verified 460b3c35

Update ci_build.sh

srayasam-amd committed 315 days ago

Verified f72eb95f

Update Dockerfile.manylinux_2_28_x86_64.rocm

srayasam-amd committed 315 days ago

Verified a3d34818

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

Ruturaj4 committed 317 days ago

Verified 39f7db68

[ROCm] Implement RNN support (#217)

Ruturaj4 committed 317 days ago

Verified dceb5310

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

Ruturaj4 committed 336 days ago

Verified da40f36d

Add gfx12xx archs (#205)

Ruturaj4 committed 341 days ago

Verified f99fbcb3

Older