Search results

Pallas kernel errors only with interpret

bugSomething isn't workingSomething isn't working

Status: Open.

#33260 In jax-ml/jax;

· sssshhhhhh opened on Nov 11, 2025

IPv6 support for JAX worker

bugSomething isn't workingSomething isn't working

Status: Open.

#33251 In jax-ml/jax;

· tyiu-google opened on Nov 11, 2025

Massive padding and fragmentation for TPU, causing OOM

bugSomething isn't workingSomething isn't working

Status: Open.

#33244 In jax-ml/jax;

· kmg3821 opened on Nov 11, 2025

[sharding] batched jnp.linalg.inv no longer works with sharding

bugSomething isn't workingSomething isn't working

Status: Open.

#33230 In jax-ml/jax;

· inailuig opened on Nov 10, 2025

conv_general_dilated fails on older GPUs (compute cap < 7.5)

bugSomething isn't workingSomething isn't working

Status: Open.

#33224 In jax-ml/jax;

· ViliamVadocz opened on Nov 10, 2025

Support for Attention Sinks on GPU in jax.nn.dot_product_attention

enhancementNew feature or requestNew feature or request

Status: Open.

#33200 In jax-ml/jax;

· Guido1Alessandro1Trevisan opened on Nov 9, 2025

Make <code>tree_flatten</code> and <code>tree_unflatten</code> dunders

enhancementNew feature or requestNew feature or request

Status: Open.

#33199 In jax-ml/jax;

· purepani opened on Nov 9, 2025

Scan suppresses runtime error handled with pure callback

bugSomething isn't workingSomething isn't working

Status: Open.

#33147 In jax-ml/jax;

· johannahaffner opened on Nov 6, 2025

Jetson Thor

bugSomething isn't workingSomething isn't working

type:support

Status: Open.

#33133 In jax-ml/jax;

· johnnynunez opened on Nov 6, 2025

jaxlib build issue on macOS Tahoe

bugSomething isn't workingSomething isn't working

Status: Open.

#33112 In jax-ml/jax;

· adamjstewart opened on Nov 5, 2025

[bug]pericision decrease when update jax from jax==0.6.2 to jax==0.8.0

bugSomething isn't workingSomething isn't working

Status: Open.

#33111 In jax-ml/jax;

· pathfinder-pf opened on Nov 5, 2025

Code does not terminate when using jax.scipy.special.sici together with jax.numpy.linalg.solve

bugSomething isn't workingSomething isn't working

Status: Open.

#33081 In jax-ml/jax;

· Pvjesper opened on Nov 4, 2025