Search results

❓Question: I cannot run jupyter lab with JAX docker image.

Status: Open.

#1673 In NVIDIA/JAX-Toolbox;

· donhuvy opened on Sep 13, 2025

jax:axlearn builds with python 3.12, needs lower python

Status: Open.

#1456 In NVIDIA/JAX-Toolbox;

· dmarx opened on May 16, 2025

Flag xla_gpu_enable_address_computation_fusion was renamed to xla_gpu_enable_dynamic_slice_fusion

Status: Open.

#1098 In NVIDIA/JAX-Toolbox;

· qGentry opened on Oct 13, 2024

Register-level FLOPS counting in JAX programs ?

Status: Open.

#836 In NVIDIA/JAX-Toolbox;

· mitkotak opened on May 18, 2024

Provide versioned tags instead of date tags

Status: Open.

#717 In NVIDIA/JAX-Toolbox;

· samos123 opened on Apr 13, 2024

Multi-cluster support in CI

Status: Open.

#616 In NVIDIA/JAX-Toolbox;

· yhtang opened on Mar 5, 2024

Report SLURM runner status as launch-slurm-runner job outcome

Status: Open.

#540 In NVIDIA/JAX-Toolbox;

· yhtang opened on Feb 14, 2024

Unique name and logfile for SLURM-launched A100 runners

Status: Open.

#525 In NVIDIA/JAX-Toolbox;

· yhtang opened on Feb 8, 2024

Add a test to check that NVCC is used for CUDA and Clang is used for pure host compilation

Status: Open.

#515 In NVIDIA/JAX-Toolbox;

· andportnoy opened on Feb 5, 2024

[Feature Request] Need Matmul Attention layer instead of Einsum to support GPU running

Status: Open.

#502 In NVIDIA/JAX-Toolbox;

· MoFHeka opened on Jan 27, 2024

Warn users not to run scripts off /opt?

Status: Open.

#475 In NVIDIA/JAX-Toolbox;

· yhtang opened on Jan 16, 2024

Add stable tag to all upstream containers

Status: Open.

#474 In NVIDIA/JAX-Toolbox;

· yhtang opened on Jan 16, 2024