Search results

jnp.log/log2/log10 return -inf for subnormal float32 inputs (XLA DAZ flush)

bugSomething isn't workingSomething isn't working

Status: Open.

#37670 In jax-ml/jax;

· Theaamadu opened on May 14, 2026

jax crashes when a CUDA enabled version is installed, but CUDA_VISIBLE_DEVICES=''

bugSomething isn't workingSomething isn't working

Status: Open.

#37668 In jax-ml/jax;

· ryanhausen opened on May 14, 2026

ci/envs/*.env files unconditionally overwrite environment variables, preventing customization

bugSomething isn't workingSomething isn't working

Status: Open.

#37662 In jax-ml/jax;

· ashahba opened on May 14, 2026

jax.lax.ragged_dot cannot be exported via jax.export for the cuda platform with a symbolic group dimension

bugSomething isn't workingSomething isn't working

Status: Open.

#37659 In jax-ml/jax;

· n-gao opened on May 14, 2026

jax.experimental.sparse.bcoo._bcoo_dot_general_transpose returns a broadcasted cotangent for vmapped sparse matvecs

bugSomething isn't workingSomething isn't working

Status: Open.

#37647 In jax-ml/jax;

· vboussange opened on May 14, 2026

Incorrect gradient of lax.digamma for negative half-integer inputs

bugSomething isn't workingSomething isn't working

Status: Open.

#37635 In jax-ml/jax;

· ALinrunrun opened on May 13, 2026

Incorrect jacfwd result for logsumexp with zero b coefficient

bugSomething isn't workingSomething isn't working

Status: Open.

#37633 In jax-ml/jax;

· ALinrunrun opened on May 13, 2026

Potential issue with shard_map + ordered IO callbacks that throw exceptions

bugSomething isn't workingSomething isn't working

Status: Open.

#37587 In jax-ml/jax;

· jburnim opened on May 12, 2026

CPU compile breaks JAX profiler writes TPU HloProtos to /host:metadata

bugSomething isn't workingSomething isn't working

Status: Open.

#37533 In jax-ml/jax;

· jaustin-ant opened on May 11, 2026

# [Perf] XLA fails to fuse segmented reductions in softmax (segment_max → exp → segment_sum), 8-19x slower than hand-fused Triton

bugSomething isn't workingSomething isn't working

Status: Open.

#37511 In jax-ml/jax;

· Umang-projects opened on May 8, 2026

CPU performance regression

bugSomething isn't workingSomething isn't working

Status: Open.

#37465 In jax-ml/jax;

· clemisch opened on May 7, 2026

Closing over numpy array constant with JAX_USE_SIMPLIFIED_JAXPR_CONSTANTS=True crashes while lowering

bugSomething isn't workingSomething isn't working

Status: Open.

#37445 In jax-ml/jax;

· bzh-bzh opened on May 7, 2026