Search results

<code>ArrayTest.testThreadsafeIndexing</code> fails under <code>pmap_shmap_merge=True</code>.

bugSomething isn't workingSomething isn't working

Status: Open.

#31911 In jax-ml/jax;

· danielsuo opened on Sep 18, 2025

Different output between <code>pmapped_f(x)</code> and <code>pmapped_f.lower(x).compile()(x)</code> under <code>pmap_shmap_merge=True</code>.

bugSomething isn't workingSomething isn't working

Status: Open.

#31910 In jax-ml/jax;

· danielsuo opened on Sep 18, 2025

Different cuda versions specified in pip install command between github and docs

bugSomething isn't workingSomething isn't working

Status: Open.

#31907 In jax-ml/jax;

· Nebulised opened on Sep 18, 2025

NotImplementedError: unreduced rule for transpose is not implemented

bugSomething isn't workingSomething isn't working

Status: Open.

#31895 In jax-ml/jax;

· batterseapower opened on Sep 18, 2025

Documentation request for hijax / lojax

enhancementNew feature or requestNew feature or request

Status: Open.

#31892 In jax-ml/jax;

· johannahaffner opened on Sep 17, 2025

Unnecessary <code>all_reduce</code> introduced to unary einsum

bugSomething isn't workingSomething isn't working

type:Bug

Status: Open.

#31889 In jax-ml/jax;

· batterseapower opened on Sep 17, 2025

<code>output_shardings</code> of compiled function has wrong dimensionality

bugSomething isn't workingSomething isn't working

type:Bug

Status: Open.

#31888 In jax-ml/jax;

· batterseapower opened on Sep 17, 2025

Unimplemented primitive in Pallas GPU lowering: scatter.

bugSomething isn't workingSomething isn't working

Status: Open.

#31876 In jax-ml/jax;

· chaoming0625 opened on Sep 17, 2025

Jax distributed deadlock

bugSomething isn't workingSomething isn't working

Status: Open.

#31870 In jax-ml/jax;

· jwnys opened on Sep 17, 2025

Wrong shape for jax.scipy.idctn

bugSomething isn't workingSomething isn't working

Status: Open.

#31836 In jax-ml/jax;

· unalmis opened on Sep 16, 2025

Aborts due to timeouts when evaluating a variable-runtime function in parallel

bugSomething isn't workingSomething isn't working

Status: Open.

#31818 In jax-ml/jax;

· gerlero opened on Sep 15, 2025

<code>JaxStackTraceBeforeTransformation</code> error for <code>odeint</code> with sparse BCOO matrix

bugSomething isn't workingSomething isn't working

Status: Open.

#31792 In jax-ml/jax;

· refraction-ray opened on Sep 13, 2025