Search results

Please build against ILP64 so Eigh (LAPACK dsyevd) can be run on matrices larger than 32766x32766

enhancementNew feature or requestNew feature or request

Status: Open.

#35919 In jax-ml/jax;

· Joshuaalbert opened on Mar 16, 2026

jax.nn.dot_product_attention silently produces incorrect results on TPU v6e

bugSomething isn't workingSomething isn't working

Status: Open.

#35916 In jax-ml/jax;

· MoreiraP12 opened on Mar 15, 2026

Native implementation of linear-crossentropy-loss with gradients

enhancementNew feature or requestNew feature or request

Status: Open.

#35906 In jax-ml/jax;

· pillow37 opened on Mar 14, 2026

jax.distributed.initialize() hangs on TPU v6e-16 multislice since 0.8.1

Status: Open.

#35905 In jax-ml/jax;

· seongmoon729 opened on Mar 14, 2026

sharded x64 arange gives wrong results

bugSomething isn't workingSomething isn't working

Status: Open.

#35896 In jax-ml/jax;

· Findus23 opened on Mar 13, 2026

Crash with multi-device reshape + vmap + jacrev

bugSomething isn't workingSomething isn't working

Status: Open.

#35815 In jax-ml/jax;

· vladfi1 opened on Mar 11, 2026

Empty bitcode string provided for eigen

bugSomething isn't workingSomething isn't working

Status: Open.

#35800 In jax-ml/jax;

· limamau opened on Mar 11, 2026

Discrepancy between jit and lower+compile

bugSomething isn't workingSomething isn't working

Status: Open.

#35799 In jax-ml/jax;

· brianwa84 opened on Mar 11, 2026

jnp.sum of all-True boolean array returns wrong result under jax.jit on GPU

Status: Open.

#35762 In jax-ml/jax;

· ettrickshepherd opened on Mar 10, 2026

Delay Kernel time out warning

bugSomething isn't workingSomething isn't working

Status: Open.

#35751 In jax-ml/jax;

· adam-hartshorne opened on Mar 10, 2026

Mosaic TPU: tpu.strided_load not implemented for stride != 1, but Pallas API accepts strided slices

Status: Open.

#35715 In jax-ml/jax;

· norx1991 opened on Mar 9, 2026

Add an unroll parameter to while_loop

enhancementNew feature or requestNew feature or request

Status: Open.

#35702 In jax-ml/jax;

· carlosgmartin opened on Mar 7, 2026