Search results

Does JAX need a deterministic reasoning kernel for emerging LLM-based decision systems?

enhancementNew feature or requestNew feature or request

Status: Open.

#33796 In jax-ml/jax;

· yuer-dsl opened on Dec 8, 2025

RuntimeError with from_dlpack(copy=True)

bugSomething isn't workingSomething isn't working

Status: Open.

#33790 In jax-ml/jax;

· haohuanw opened on Dec 8, 2025

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