All issues

[Pallas TPU] Pallas MLIR lowering cache incorrectly reuses source locations across operations

#41153

· ayaka14732 opened

on Sep 30, 2026

`lax.pmean` with unequal `axis_index_groups` divides every group by the size of the first group

#41132

· the-Shallow opened

on Sep 30, 2026

jax.fwd_and_bwd silently discards keyword arguments

#41121

· the-Shallow opened

on Sep 29, 2026

jax.numpy.random.multinomial shape behavior is inconsistent with docs, similar functions, and numpy

#41107

· stroxler opened

on Sep 29, 2026

Error lowering pallas `fori_loop(unroll=True)` with `pltpu.ACC`

#41104

· skye-ant opened

on Sep 29, 2026

jnp.modf returns nan as the fractional part for +/-inf (numpy returns +/-0)

#41102

· jayzuccarelli opened

on Sep 29, 2026

Persistent compilation cache key depends on the cache directory's path when the XLA kernel cache is enabled

#41044

· tsvikas opened

on Sep 28, 2026

jnp.linalg.svd silently returns NaN on ill-conditioned float64 with nvidia-cusolver 12.2.6.9 (gesvdj)

#41042

· yingjerkao opened

on Sep 28, 2026

jax.experimental.jet: Bugs for zero-arg primitives (e.g. iota) and in series scaling for discrete primitives (e.g. clz, ...)

#41041

· paulgekeler opened

on Sep 28, 2026

CPU: jitted lax.scan over a chunked linear-attention layer returns wrong values in some processes

#41025

· AshishKumar4 opened

on Sep 28, 2026

jax.export serialization loses weak_type and changes dtype promotion

#41007

· PiesAvalon opened

on Sep 26, 2026

Batched complex64 `jax.ops.segment_sum`/`scatter_add` returns wrong values and faults with CUDA_ERROR_ILLEGAL_ADDRESS on GPU

#40970

· gRox167 opened

on Sep 25, 2026