Search results

MacOS: Some <code>mujoco.mjx</code> models are &gt;100x slower on <code>jax&gt;=0.7</code> than on <code>jax&lt;0.7</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#33761 In jax-ml/jax;

· hartikainen opened on Dec 5, 2025

jax.errors.JaxRuntimeError: INTERNAL: Autotuning failed for HLO

bugSomething isn't workingSomething isn't working

Status: Open.

#33715 In jax-ml/jax;

· geligeli opened on Dec 4, 2025

SVD does not propagate NaNs for batch sizes &gt;2

bugSomething isn't workingSomething isn't working

Status: Open.

#33707 In jax-ml/jax;

· amacati opened on Dec 3, 2025

Device -&gt; host transfer causes CUDA OOM on WSL2

bugSomething isn't workingSomething isn't working

Status: Open.

#33694 In jax-ml/jax;

· jkyl opened on Dec 3, 2025

JAX encounters a core dump error

bugSomething isn't workingSomething isn't working

Status: Open.

#33679 In jax-ml/jax;

· Henry0132 opened on Dec 3, 2025

JAX project runs faster with JAX 0.4.28 than any later version

Status: Open.

#33666 In jax-ml/jax;

· cgiovanetti opened on Dec 2, 2025

'memory_kind' is not respected in jax.export while using jax.shard_map

bugSomething isn't workingSomething isn't working

Status: Open.

#33657 In jax-ml/jax;

· Vikisix opened on Dec 2, 2025

'pinned_host' memory kind is not keept after jax export.

bugSomething isn't workingSomething isn't working

Status: Open.

#33649 In jax-ml/jax;

· Vikisix opened on Dec 2, 2025

Inefficient concatenate (many tiny NCCL kernels)

bugSomething isn't workingSomething isn't working

Status: Open.

#33646 In jax-ml/jax;

· yhtang opened on Dec 2, 2025

Support <code>Layout.tiling</code> in <code>with_layout_constraint</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#33543 In jax-ml/jax;

· yuwei-qin opened on Nov 25, 2025

Complex casting warning not needed

bugSomething isn't workingSomething isn't working

Status: Open.

#33521 In jax-ml/jax;

· inversecrime opened on Nov 25, 2025

<code>jax.experimental.sparse.CSR</code> accepts wrong shapes

bugSomething isn't workingSomething isn't working

Status: Open.

#33514 In jax-ml/jax;

· johannahaffner opened on Nov 24, 2025