Search results

integer overflow in <code>jnp.clip</code> with unsigned ints

bugSomething isn't workingSomething isn't working

Status: Open.

#34455 In jax-ml/jax;

· ev-br opened on Jan 17, 2026

Numerical Discrepancies between Pallas Kernel and Standard JAX with bfloat16

bugSomething isn't workingSomething isn't working

Status: Open.

#34454 In jax-ml/jax;

· depksingh opened on Jan 17, 2026

Facing Runtime error while running kernel but works with interpret mode

bugSomething isn't workingSomething isn't working

Status: Open.

#34431 In jax-ml/jax;

· depksingh opened on Jan 16, 2026

jaxlib nightly not installed with --pre

bugSomething isn't workingSomething isn't working

Status: Open.

#34421 In jax-ml/jax;

· guptaaka opened on Jan 15, 2026

Unsupported on ROCm Misc tests

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34388 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/svd_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34386 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/string_array_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34384 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/sparse_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34381 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/shard_map_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34380 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/random_lax_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34378 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/polynomial_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34377 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026

Unsupported ROCm Tests <code>tests/optimizers_test.py</code>

AMD GPUIssues pertaining to AMD GPUs (ROCM)Issues pertaining to AMD GPUs (ROCM)

bugSomething isn't workingSomething isn't working

Status: Open.

#34376 In jax-ml/jax;

· JehandadKhan opened on Jan 14, 2026