Search results

XLA topk-decomposer crash in pmap(vmap(fn)) with top_k[0] in jnp.where + separate top_k[1]

type:Bug

Status: Open.

#36703 In jax-ml/jax;

· jz-sa opened on Apr 13, 2026

jax.export.deserialize().call() fails with internal XLA error for shard_map across 2x2 mesh

bugSomething isn't workingSomething isn't working

Status: Open.

#36691 In jax-ml/jax;

· nickmcgreivy opened on Apr 11, 2026

D2H memcpy is followed by a Waiting for Host Callback Execution stall of 16 ms

bugSomething isn't workingSomething isn't working

Status: Open.

#36651 In jax-ml/jax;

· JulesGuillou opened on Apr 9, 2026

XLA numerical inconsistency

bugSomething isn't workingSomething isn't working

Status: Open.

#36625 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

numerical inconsistency

bugSomething isn't workingSomething isn't working

Status: Open.

#36624 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

JAX/XLA produces numerically unstable results for normalization-heavy pipelines under near-zero inputs.

bugSomething isn't workingSomething isn't working

Status: Open.

#36623 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

jax.jit (XLA) produces wrong output

bugSomething isn't workingSomething isn't working

Status: Open.

#36622 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

Silent numerical inconsistency under compiler optimization

bugSomething isn't workingSomething isn't working

Status: Open.

#36620 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

jax.jit produces incorrect results vs eager execution on CPU

bugSomething isn't workingSomething isn't working

Status: Open.

#36619 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

jax.jit produces different results compared to eager execution

bugSomething isn't workingSomething isn't working

Status: Open.

#36618 In jax-ml/jax;

· beanduan22 opened on Apr 9, 2026

ffi_call with empty array inside lax.scan crashes during lowering when JAX_USE_SIMPLIFIED_JAXPR_CONSTANTS=True

bugSomething isn't workingSomething isn't working

Status: Open.

#36607 In jax-ml/jax;

· PTNobel opened on Apr 8, 2026

dot_product_attention is non-deterministic across sequence lengths: different KV-cache sizes produce different XLA reduction trees

Status: Open.

#36570 In jax-ml/jax;

· ry2009 opened on Apr 7, 2026