8 Pull requests merged by 4 people

• JAX package path is SRC_PATH_JAX

  #1418 merged Apr 25, 2025

• Ensure JAX installation is editable

  #1416 merged Apr 25, 2025

• Fix TE build broken by #1409

  #1415 merged Apr 24, 2025

• nsys-jax: explicitly synchronise before cudaProfilerStop in tests

  #1412 merged Apr 24, 2025

• Skip the unversioned symlink in cuDNN >= 9.10

  #1411 merged Apr 24, 2025

• Fix redundant install of JAX wheels

  #1409 merged Apr 23, 2025

• Dockerfile.base: default value for JAX_TOOLBOX_REF

  #1407 merged Apr 22, 2025

• Infer compute cap from CUDA_ARCH_LIST if possible

  #1406 merged Apr 21, 2025

2 Unresolved conversations

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Run NCCL tests on the JAX-specific base container

  #1284 commented on Apr 23, 2025 • 0 new comments

• test tree_util changes

  #1396 commented on Apr 24, 2025 • 0 new comments