Go
Home
Pricing
FAQ
Install
Home
Pricing
FAQ
Install
Login
via GitHub
ROCm/jax
Pull Requests
Commits
fix_test_shard
0.5.0-fix_build_clang19_tmp
5.0_skiptests
Ruturaj4-qa-dockerfile-changes
abort_detection_ci_workflow_gulsum
add_pytest_rocm_workflow
add_support_of_building_jax_targets_under_an_umbrella_workspace
add_support_of_building_jax_under_umbrella_workspace_new
add_support_of_building_jax_under_umbrella_workspace_new2
add_support_of_building_jax_under_umbrella_workspace
add_support_to_dynamic_set_of_test_deps
add-device-test-v0.8.2
add-device-test-v0.9.0
add-rocm-device-placement-compat-test
add-rocm-lu-backward-compat-test
add-rocm-lu-pivots-to-permutation-test
add-run-bazel-test-rocm-rbe
add-run-bazel-test-rocm-rbe-0.8.2
amd-main
amd-main-ci-matrix-test
amdgpu_gfx950_support
ammallya-0.6.0
arati/fix_hipblas_resource_issue
arech_mosaic00_upstream
arech_mosaic00_upstream01_tmp
arech_mosaic00_upstream01
arech_mosaic00
arech_mosaic00_082
arech_mosaic00_090
bring_rocm_dlm_perf
build-fix-changes
build-fixes-rollup
ci_absl_status
ci_add_block_pingpong_v2
ci_add_block_pingpong_v3
ci_add_new_gfx-upstream
ci_add_packages_dockerfile
ci_add_triton_extension
ci_apt_update
ci_bring_up_clang
ci_bring_up_clang_30_qa
ci_bring_up_rnn_support
ci_build_code_fixes
ci_build_code_fixes-upstream
ci_build_fix
ci_build_fix_2
ci_cj_fix_rocm-jaxlib-v0.7.1
ci_cj_profiler_test_rocm-jaxlib-v0.8.0
ci_cj_profiler_test_rocm-jaxlib-v0.9.0
ci_cj_rocm_profiler_test
ci_cj_rocprofv3-rocm-jaxlib-v0.5.0
ci_cj_rocprofv3-rocm-jaxlib-v0.6.0
ci_cj_slogdet_small_rocm-jaxlib-v0.8.2
ci_cj_slogdet-fix
ci_clang
ci_dockerfile_arg_changes
ci_dockerfile_arg_changes-upstream
ci_enable_https
ci_enable_https-upstream
ci_export_test_fix
ci_f8_tests
ci_f8
ci_fix_jax_rocm_plugin_deps
ci_fix_multi_gpu_test_logic-upstream
ci_fix_rnn_lowering
ci_fix_rnn_lowering-upstream
ci_fix_set_options
ci_fix_set_options-upstream
ci_fix_solver_paths
ci_fix_wheelhouse_relative_paths-upstream
ci_generate_csv
ci_hybrid_typo-upstream
ci_install_sys_libs
ci_install_sys_libs-upstream
ci_jax_export_harness
ci_jax_pjrt
ci_kernel_launch_dims_fix
ci_linter_fix-upstream
ci_matrix_test_kernel_serialize
ci_matrix_test_nosafepythonpath
ci_matrix_test_shmem_upgrade
ci_matrix_test
ci_multiprocess_gpu_test
ci_op_json_csv
ci_pjrt
ci_rnn_final
ci_rnn_final-upstream
ci_rnn_rv
ci_rocm_readme
ci_rocm_version_fix
ci_rv_clang_clean
ci_rv_clang
ci_rv_mt
ci_sparse_matmul_test
ci_sparse_qr_test
ci_spmm
ci_test_harness_vmap
ci_typed_xla_ffi
ci_typename
ci-upstream-sync-12_1
ci-upstream-sync-23_1
ci-upstream-sync-55_1
ci-upstream-sync-67_1
ci-upstream-sync-109_1
ci-upstream-sync-128_1
ci-upstream-sync-149_1
ci-upstream-sync-152_1
ci-upstream-sync-153_1
ci-upstream-sync-154_1
ci-upstream-sync-155_1
ci-upstream-sync-156_1
ci-upstream-sync-166_1
ci-upstream-sync-167_1
ci-upstream-sync-168_1-reb
ci-upstream-sync-168_1
ci-upstream-sync-169_1
ci-upstream-sync-170_1
ci-upstream-sync-171_1
ci-upstream-sync-172_1
ci-upstream-sync-173_1
ci-upstream-sync-183_1
ci-upstream-sync-185_1
ci-upstream-sync-186_1
ci-upstream-sync-187_1
ci-upstream-sync-188_1
ci-upstream-sync-189_1
ci-upstream-sync-190_1
ci-upstream-sync-191_1
ci-upstream-sync-192_1
ci-upstream-sync-193_1
ci-upstream-sync-215_1
ci-upstream-sync-216_1
ci-upstream-sync-217_1
ci-upstream-sync-218_1
ci-upstream-sync-220_1
ci-upstream-sync-221_1
ci-upstream-sync-222_1
ci-upstream-sync-223_1
ci-upstream-sync-224_1
ci-upstream-sync-225_1
ci-upstream-sync-226_1
ci-upstream-sync-227_1
ci-upstream-sync-228_1
ci-upstream-sync-229_1
ci-upstream-sync-230_1
ci-upstream-sync-231_1
ci-upstream-sync-232_1
ci-upstream-sync-233_1
ci-upstream-sync-234_1
ci-upstream-sync-235_1
ci-upstream-sync-236_1
ci-upstream-sync-237_1
ci-upstream-sync-238_1
ci-upstream-sync-239_1
ci-upstream-sync-240_1
ci-upstream-sync-241_1
ci-upstream-sync-242_1
ci-upstream-sync-243_1
ci-upstream-sync-244_1
ci-upstream-sync-245_1
ci-ut-artifacts
cj_rocm-jaxlib-v0.5.0-waves_per_eu-2
cleanup-gpu-targets
continue_on_fail_test
cp-test-pmap
cph-bazel-unit-tests
create_new_patch
csrlsvqr-support-v0.8.2
disable-inline-asm-test-jaxlib-v0.8.0
enabl-upstream-ci-on-release
enable_random_lax_test
enable-ToeplitzSym-tests-jaxlib-v0.8.0
enable-linalg-cond-tests-v0.8.0
enable-pallas-interpret-rocm
enable-rnn-workspace-overflow-test-v0.8.0
enable-rsf2csf-tests-v0.8.0
enable-tridiagonal-reduction-tests-jaxlib-v0.8.0
feat/manylinux_2_28
find-pyversion-at-build-time
find-pyversion-at-build-time-0.5.0
fix/bitshift_unit_test
fix_ci_linux_distro
fix_cudnn_rocm
fix_handle_pool
fix_hash_5.0_0
fix_hipblas_35-qa
fix_pallas_bitwise_left_shift_test
fix/pallas-tests-shared-memory
fix_psend_tests
fix_rnn_struct_encoding_cherry_pick
fix_rnn_struct_encoding_upstream
fix_rnn_struct_encoding_090_cherry_pick
fix/rocm7_plugin_dependency
fix/shard_map_test_failures
fix_test_shard
fix_unary_primitives
fix-auditwheel
fix-auditwheel-version-0.5.0
fix-cleanup-race-condition
fix-el8-repo-urls
fix-lax-backed-scipy-tests
fix-rnn-struct-encoding-v0.8.0
fix-rocm-pallas-lowerings
fix-subdtype-test
fix-test-pallas-tests-gfx11xx-gfx12xx
fix-wheel-version-constraint
fixes-bwd-fused-attention-unit-tests
fixes-bwd-fused-attention-unit-tests-0.6.0
fixes-for-0.5.0-build
float-and-complex-builtins-error
force_true_cse
genesu/build-jax-0.7.1
genesu/make-nvidia-arg-optional
geqrf-compat-test-v0.8.2
gesvd-compat-test-v0.8.2
gesvdj-support-jaxlib-v0.8.0
gesvdj-support-jaxlib-v0.8.2
gfx950_support
gh-9948-add-gpu-ci-upstream
gmap
gpu-determinism-note
gtsv2-support-jax-v0.8.0
gulsum_ci_matrix_test
handle-dunder-array-classes
hessian-docs-iteration
hmm-example
hoist-consts
ignore-warnings-jax2tf-tests
improve-tracer-error
introduce_rocm_jax_pipeline
issue744
issue768
issue831
issue1236
issue1297
issue1486
jax_UT_tridiagonal_for_rocm-upstream
jax_pypi
jax_pypi-upstream
jax_ut_pallas_mha_bwd
jax-triton
jaxlib-build-changes
jaxlib-v0.4.18-rocm-6.0-batra
jb/sinc-double-where
jb/staxperiments
jb/staxperiments-wip
jb/tagging
jd/gpu_ci
jd/plugin_ver_upstream
jd/rocm_version_skip
jd/wsl_experimental_rebased
jd/wsl_experimental
jd-435-xla-upd
jd-gfx1101-upstream
jet
jet2
jit-of-multi-host-pmap
jvadakat_jax_patch
kithumma/jaxlib-rocmversion-therock-tag
kithumma/therock-tag-jaxlib-0.8.0
lax_dot_fail
lax-ad-test-skip
lax-api-boundary
lazy-simplification-2
lazy-sublanguage-eager-fusion
linear-transpose
linearize
log-resharding
logging-abort
long-line
main
main-old
make-custom-vjp-bwd-nones-more-robust
mapped-invars-fixes
mark-mosaics-xfail
masking-revisions
min_max_atomics
minor-source-sync
multi-host-pmap-fix
multivariate-normal-logpdf-improvements
new_ci/v1
no-gcc-download
numpy-dispatch-test-count
ode-test
ode-tests
okakarpa-patch-1
oldmain
omnistaging-doc-fix-links
omnistaging-multihost
padding-tuple
pallas_mha_opt_v2
pallas_mha_opt
parallel
pdot
phambinh/add_copy_build_pattern_rocm
phambinh/add_hip_runtime_hybird
phambinh/approx-tanh-rocm-impl
phambinh/backup_pool
phambinh/ci_fix_build_nvidia
phambinh/ci_rocm_approx_tanh
phambinh/ci_rocm_misc_tests
phambinh/ci_rocm_nn_tests
phambinh/ci_rocm_pallas_ops_tests
phambinh/ci_rocm_scaled_matmul_tests
phambinh/ctran-collective-backend
phambinh/fix_build_nvidia
phambinh/fix_ffi
phambinh/fix_gpu_paged_testcases
phambinh/fix_gpu_paged_8
phambinh/fix_mismatch_spare_api
phambinh/fix_testcase11
phambinh/fix_vmap
phambinh/get_gpu_number_rocm-smi
phambinh/handle-pool-singleton
phambinh/handle-pool-singleton-separation
phambinh/jax_fix_copy
phambinh/nn-rocm-skip-cudnn
phambinh/skip-approx-tanh-rocm
phambinh/skip-get-vmap-slice-only-bug
phambinh/skip-tests-put-stream-not-supported-jax-0.8
phambinh/unskip-attention-bias-gradient-rocm
phambinh/unskip-comparison-scalar-rocm
phambinh/unskip-comparison-scalar-rocm-v0.8.2
phambinh/unskip-layout-donation-test-rocm
phambinh/wip1
pmap
pmap-docs
pmean-fix
port-rocm-patches-v0.8.1
porting_tests_related_changes_from_0.6.0
porting_v0.6.0_changes_to_v0.7.1
pretty-print-improvements
primitives
print-partial-eval-stats
prng-slow
prngkey-linearity-brainstorming
pytype-fix
r0.5.0-fix-getrfBatched_template
ray
rbe-config
rcom-ci-tsl-path-fix
reenable-some-mypy-checks
register_rocm_to_block_scaled_dot_lowering_path
release_v2.1_rocm
release/0.4.33
remove_failure_limit_from_pytest
remove-app
remove-double-dtype-warning
remove-old-rocm-build-code
remove-prng-maxsize-restriction
remove-test
remove-unneeded-code
remove-x64-mode
removed_unsupported_gfx_arch
rename-mastertrace
reraise
revert3674
revert-568-remove-unneeded
revert-702-phambinh/skip-tests-jax-0.9.0
revert-3983-device-put-tuple
revert-4160-copybara1
revert-33757-register_rocm_to_block_scaled_dot_lowering_path
revert-squash
revised_ci_matrix_test_upstream_main
revised_ci_matrix_test
revive-leak-checker
rocm_add_hipStreamWaitEvent
rocm_dockerfile_and_test_updates
rocm_enable_tridiagonal_tests
rocm_fix_pallas
rocm_gesvdj_support
rocm_gpu_lowering_gesvdj
rocm_no_cache
rocm_pallas_support
rocm_psend_support
rocm_test_script_updates
rocm7-plugin-updates
rocm-add-device-test
rocm-add-triton_command_buffer
rocm-build-updates-rocm5.7
rocm-cholesky-solver-potrf
rocm-dev-build-fixes
rocm-eigh-export-compat
rocm-enable_tridiagonal_solve
rocm-enable-cond-toeplitz-tests
rocm-enable-triton-pallas-tests
rocm-enable-workspace-overflow-test
rocm-export_test-add-rocm-platform
rocm-fix-numalib
rocm-geqrf-compat-test
rocm-gesvd-compat-test
rocm-hip-threefry-ffi-stable
rocm-jax-stable-2024_05_20
rocm-jax-stable-2024_05_27
rocm-jax-stable-2024_06_03
rocm-jax-stable-2024_07_01
rocm-jax-stable-2024_08_07
rocm-jax-stable-2024_08_15
rocm-jax-stable-2024_08_19
rocm-jax-triton
rocm-jax-triton-add-get_arch_detail
rocm-jaxlib-docker-rocm6.0-updates
rocm-jaxlib-v0.4.3
rocm-jaxlib-v0.4.4
rocm-jaxlib-v0.4.6-rocm55
rocm-jaxlib-v0.4.6
rocm-jaxlib-v0.4.9-rocm56
rocm-jaxlib-v0.4.9-w-nccl-bf16-fix
rocm-jaxlib-v0.4.9
rocm-jaxlib-v0.4.10
rocm-jaxlib-v0.4.11
rocm-jaxlib-v0.4.12-hipblaslt
rocm-jaxlib-v0.4.12-jax-triton
rocm-jaxlib-v0.4.12-jax-triton-rocm55
rocm-jaxlib-v0.4.12
rocm-jaxlib-v0.4.13-jax-triton
rocm-jaxlib-v0.4.13-jax-triton-rocm56
rocm-jaxlib-v0.4.13-perf-gpt-j
rocm-jaxlib-v0.4.13-rocm56
rocm-jaxlib-v0.4.13
rocm-jaxlib-v0.4.14-rocm56
rocm-jaxlib-v0.4.14
rocm-jaxlib-v0.4.15
rocm-jaxlib-v0.4.16
rocm-jaxlib-v0.4.17
rocm-jaxlib-v0.4.18-attn-update
rocm-jaxlib-v0.4.18
rocm-jaxlib-v0.4.19
rocm-jaxlib-v0.4.20-jax-triton
rocm-jaxlib-v0.4.20-rocm6.0-jax-triton
rocm-jaxlib-v0.4.20-rocm6.0-jax-triton-attn-transpose
rocm-jaxlib-v0.4.20-rocm6.0
rocm-jaxlib-v0.4.20-w-sparse-fix
rocm-jaxlib-v0.4.20
rocm-jaxlib-v0.4.21
rocm-jaxlib-v0.4.23-cqe
rocm-jaxlib-v0.4.23-idass
rocm-jaxlib-v0.4.23-idass-SQUASHED
rocm-jaxlib-v0.4.23-kithumma_test
rocm-jaxlib-v0.4.23-pytest_report
rocm-jaxlib-v0.4.23-squashed
rocm-jaxlib-v0.4.23-tridiagonal-tests
rocm-jaxlib-v0.4.23-w-profiler-fixes-rbatra
rocm-jaxlib-v0.4.23
rocm-jaxlib-v0.4.24-jax-triton
rocm-jaxlib-v0.4.24
rocm-jaxlib-v0.4.25-jax-triton
rocm-jaxlib-v0.4.25
rocm-jaxlib-v0.4.26
rocm-jaxlib-v0.4.27
rocm-jaxlib-v0.4.28-kithumma
rocm-jaxlib-v0.4.28-pallas
rocm-jaxlib-v0.4.28-qa
rocm-jaxlib-v0.4.28
rocm-jaxlib-v0.4.29
rocm-jaxlib-v0.4.30-qa_wheelhouse_fix
rocm-jaxlib-v0.4.30-qa
rocm-jaxlib-v0.4.30-wheelhouse-fix
rocm-jaxlib-v0.4.30
rocm-jaxlib-v0.4.31-qa
rocm-jaxlib-v0.4.31
rocm-jaxlib-v0.4.32
rocm-jaxlib-v0.4.33
rocm-jaxlib-v0.4.34
rocm-jaxlib-v0.4.35-qa
rocm-jaxlib-v0.4.35-qa-fix
rocm-jaxlib-v0.4.35
rocm-jaxlib-v0.4.38
rocm-jaxlib-v0.5.0-fix_build_clang_19
rocm-jaxlib-v0.5.0_fix_testEighTinyNorm
rocm-jaxlib-v0.5.0
rocm-jaxlib-v0.6.0
rocm-jaxlib-v0.6.2-bringup
rocm-jaxlib-v0.6.2
rocm-jaxlib-v0.7.0
rocm-jaxlib-v0.7.1
rocm-jaxlib-v0.8.0_mx_datatype
rocm-jaxlib-v0.8.0
rocm-jaxlib-v0.8.1-old
rocm-jaxlib-v0.8.1
rocm-jaxlib-v0.8.2
rocm-jaxlib-v0.9.0
rocm-jaxlib-weekly_24_01_25-f67d48234802e50e851618029ca76ea934403db6
rocm-jaxlib-weekly_24_01_31-b405ce7f379d8745d653625747030a02b083f351
rocm-jaxlib-weekly-04_08_24
rocm-log-pytest-html-report
rocm-rpath-direct-adaptions
rocm-serial-compat-test
rocm-skip-prim-tri-solve-test
rocm-sparse-fix
rocm-sytrd-compat-test
rocm-threefry2x32-test
rocm-wheels-0.4.35
run-less-rocm-tests
run-pytest-rocm
runner-test
rv_stable_051624
serial-compat-test-downstream-v0.8.2
serial-compat-test-downstream-v0.9.0
serial-compat-test-v0.9.0
set_release_rpaths_to_rocm_so_targets
side-effects
simpler-select-jvp
simplify_jax_wheel_build
skip_additional_tests
skip_batch_sharding_upstream
skip_tridiagonal_solve_rocm
skip_tridiagonal_solve
skip-change-rocm-080-2
skip-csr-mat-tests-rocm63
skip-dlpack-tests-on-jaxlib-version
skip-eigh-tiny-norm
skip-eigh-tiny-norm-upstream
skip-eigh-tiny-norm-v0.8.2-clean
skip-eigh-tiny-norm-v0.8.2
skip-eigh-tiny-norm-v0.9.0
skip-inline-asm-test-v0.8.0
skip-jvp13-upstream-main
skip-jvp13-v0.8.0
skip-jvp13-v0.8.2-clean
skip-jvp13-v0.8.2
skip-prim-tri-solve-test-v0.8.0
skip-prim-tri-solve-test-v0.8.2
skip-tridiagonal-solve-grad-upstream-main
skip-tridiagonal-solve-grad-v0.8.0
skip-tridiagonal-solve-grad-v0.8.2
skip-tridiagonal-solve-grad-v0.9.0
software-pipeline
source-line-info-experiments
speed-up-hips
srayasam/test
srayasam_test
stack-traces
static-argnums-immutable
stax-bias-revert
staxperiments
subodube/jaxlib-rocm-version-extra
support_wheels_as_build_deps
svd-gpu-lowering-fix-jaxlib-v0.8.0
switch_to_xla_from_release_branch
switch-grad-weak-type-test
sytrd-compat-test-v0.8.2
tb-util-path
test_333130052
test_336082045
test_336130333
test_336305193
test_336647290
test_337964891
test_338274969
test_340511784
test-amd-main
test-amd-main-artifacts
test-amd-main-with-miopen-fix
test-amd-main-with-miopen-fix-artifacts
test-docs
test-pypi-v0.9.0-comparison
test-pytest-abort-plugin
test-with-upstream-state
tf_bump
tot-rocm-skips
traceback-register
travis-conda-build-issue
trial-v0.4.36
tuple-pad-2
tuple-simplification
tweak-tolerance
tweaks
tycheck-invar-eqn-context
typo
undo-tree
unroll
unskip_multivariate_normal_rocm
unskip_tests
update_hash_35_qa
update-amd-install-docs
update-notebooks
ups-select-skip-conv
upstream_ci_download_jax_rocm_wheels
upstream_fix_handle_pool
upstream_rocm_ci
upstream
upstream-main
use-linalg-tests
use-new-runner
weekly_05-13-24
weekly-stable-rc1
while-loop-of-pmap-bug
zero-fixes2
Update tests/lax_numpy_test.py
gulsumgudukbay
committed
257 days ago
c8411cc7
Update tests/experimental_rnn_test.py
gulsumgudukbay
committed
257 days ago
f005d96c
[0.6.0-UT] skip failing tests
gulsumgudukbay
committed
257 days ago
5e534a5b
Remove warnings.warn from get_rocm_version
psanal35
committed
257 days ago
Verified
f95ad1ab
Add missing header `warnings` for test_util.py (#458)
psanal35
committed
257 days ago
Verified
7be899a0
Fix get_rocm_version to skip tests if installation is missing
psanal35
committed
257 days ago
Verified
3820e7ee
updated the XLA commit hash
gulsumgudukbay
committed
257 days ago
4694374d
Build with ROCm 7.0 on new runner (#449)
charleshofer
committed
257 days ago
Verified
fbfeba5e
[0.6.0-UT] Fix NumpyLinalgTest test cases (#453)
phambinhfin
committed
258 days ago
Verified
29c392d0
Select the platform GPUs for a few MultiProcessGpuTest test cases.
Pham Binh
committed
258 days ago
f840dc0a
Update rocm-test-requirements.txt
gulsumgudukbay
committed
258 days ago
9e5c47cb
Revert "fixed typo"
gulsumgudukbay
committed
258 days ago
69ec57e6
fixed typo
gulsumgudukbay
committed
259 days ago
d2b14cb4
[0.6.0-UT] adding a QA test requirements file for the run_single_gpu test script… (#446)
gulsumgudukbay
committed
259 days ago
Verified
1fc5d2a5
[0.6.0-UT] Skip test_dot for ROCm <6.5, and shapes exceed shared memory limits (#448)
psanal35
committed
259 days ago
Verified
d6456448
[0.6.0-UT] Add shared utility function get_rocm_version to test_util.py
psanal35
committed
260 days ago
Verified
893c907c
unskip test_select_n as it is passing on 0.6.0
gulsumgudukbay
committed
264 days ago
5b38406a
added conditional seq_len assignment for lower dimension for tests/pallas/gpu_ops_test.py::FusedAttentionTest::test_fused_attention_[fwd,bwd] tests with the rationale. The test passes now because the input fits in shared memory. before, the tests were failing due to the seq exceeding the shared memory size and causing XLARuntimeError.
gulsumgudukbay
committed
264 days ago
5659c086
[JAX] Fix numerical instability in testEighTinyNorm for ROCm GPU
phambinhfin
committed
264 days ago
3c6ab9f3
JAX release v0.6.0
hawkinsp
committed
292 days ago
9e9a3f03
[Pallas] Fix relative links in pipelining docs
justinjfu
committed
292 days ago
28538e4c
Fix bug in argnums_partial_except when static_argnums is unsorted.
dfm
committed
292 days ago
34234010
Return False in `is_env_present` if importing kubernetes leads to a ModuleNotFoundError
yashk2810
committed
292 days ago
8d3a92e6
Fix a bug in reduce_window sharding rule where padding is a tuple but we were checking for a scalar instead. Fixes https://github.com/jax-ml/jax/issues/28070
yashk2810
committed
292 days ago
33d43f7a
[Pallas] Move pipelining doc figures to static folder
justinjfu
committed
292 days ago
a9465e70
Add explicit sharding to Introduction to parallel programming doc
yashk2810
committed
292 days ago
24fdc0ad
Update XLA dependency to use revision
Google-ML-Automation
committed
292 days ago
6b94a5ce
[Pallas] Refactor pipelining docs.
justinjfu
committed
292 days ago
17e2dcaf
Update job name to show "build only" on Linux ARM64.
nitins17
committed
292 days ago
bc4fcd87
disable packed layout test on old arch prior to Hopper
Cjkkkk
committed
292 days ago
fb875101
Newer
Older