Search results

Regression in JAX 0.8.2: XLA <code>permutation_sort_simplifier</code> HLO verifier crashes compiling argsort-based inverse permutation (int64/x64)

bugSomething isn't workingSomething isn't working

Status: Open.

#34096 In jax-ml/jax;

· bmucsanyi opened on Dec 23, 2025

XLA GPU: “Contracting dimension is too fragmented” triggered by certain seq_len × batch combinations

bugSomething isn't workingSomething isn't working

Status: Open.

#34094 In jax-ml/jax;

· lych1233 opened on Dec 23, 2025

What is the replacement for <code>jax.core.mapped_aval</code>?

Status: Open.

#34087 In jax-ml/jax;

· patrick-kidger opened on Dec 22, 2025

<code>ValueError</code> "INTERNAL: CpuCallback error calling callback" rather than <code>JaxRuntimeError</code> since jax&gt;=0.7.2, depending on execution order

bugSomething isn't workingSomething isn't working

Status: Open.

#34083 In jax-ml/jax;

· bonauer-pf opened on Dec 22, 2025

[BUG] batch variant: the results of jax.lax.dot_general vary from batches.

bugSomething isn't workingSomething isn't working

Status: Open.

#34080 In jax-ml/jax;

· aolemila opened on Dec 22, 2025

PyTree data classes can have attributes registered as both <code>data_fields</code> and <code>meta_fields</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#34079 In jax-ml/jax;

· kentslaney opened on Dec 21, 2025

Update <code>jax.checkpoint</code> API to support more flexible <code>static</code> arguments like <code>jax.jit</code>

enhancementNew feature or requestNew feature or request

Status: Open.

#34070 In jax-ml/jax;

· tloen opened on Dec 19, 2025

NVIDIA GPU init failed with self built jaxlib

bugSomething isn't workingSomething isn't working

Status: Open.

#34049 In jax-ml/jax;

· gotorion opened on Dec 19, 2025

Regression in 0.8.2 for FFI interface?

Status: Open.

#34047 In jax-ml/jax;

· mariogeiger opened on Dec 19, 2025

Can't index jax.Array that's been device_put() into host memory_space

bugSomething isn't workingSomething isn't working

Status: Open.

#34041 In jax-ml/jax;

· georgevdd opened on Dec 19, 2025

Crash for CNN with a specific batch size

bugSomething isn't workingSomething isn't working

Status: Open.

#34010 In jax-ml/jax;

· cherrywoods opened on Dec 17, 2025

[mosaic] broadcast bug

bugSomething isn't workingSomething isn't working

Status: Open.

#34001 In jax-ml/jax;

· oliverdutton opened on Dec 17, 2025