Search results

Removal of legacy from_dlpack breaks TensorFlowToJax tests

bugSomething isn't workingSomething isn't working

Status: Open.

#32822 In jax-ml/jax;

· pearu opened on Oct 22, 2025

Slow down after decorating with <code>custom_vjp</code>

questionQuestions for the JAX teamQuestions for the JAX team

Status: Open.

#32802 In jax-ml/jax;

· HGangloff opened on Oct 22, 2025

Metal: failed to legalize operation 'mhlo.return' / update to StableHLO ?

Apple GPU (Metal) plugin

enhancementNew feature or requestNew feature or request

Status: Open.

#32800 In jax-ml/jax;

· janpfeifer opened on Oct 22, 2025

vjp of <code>ShapeDtypeStruct</code>: works in jax 0.7 fails on 0.8

bugSomething isn't workingSomething isn't working

Status: Open.

#32748 In jax-ml/jax;

· PhilipVinc opened on Oct 20, 2025

jaxlib from source for native GPU use on Windows without WSL2

enhancementNew feature or requestNew feature or request

Status: Open.

#32741 In jax-ml/jax;

· bbq572 opened on Oct 19, 2025

Introduce a stable/public API to query cuda version for GPU backend

enhancementNew feature or requestNew feature or request

needs infoMore information is required to diagnose & prioritize the issue.More information is required to diagnose & prioritize the issue.

NVIDIA GPUIssues specific to NVIDIA GPUsIssues specific to NVIDIA GPUs

Status: Open.

#32729 In jax-ml/jax;

· ksivaman opened on Oct 17, 2025

<code>jax.numpy.array(x)</code> chooses dtype before calling <code>__array__(dtype)</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#32725 In jax-ml/jax;

· kcdodd opened on Oct 17, 2025

About jit of multi-host pmap not implemented

bugSomething isn't workingSomething isn't working

Status: Open.

#32710 In jax-ml/jax;

· matrix97317 opened on Oct 17, 2025

Poor multithreading even for simple vectorization

bugSomething isn't workingSomething isn't working

Status: Open.

#32707 In jax-ml/jax;

· SUSYUSTC opened on Oct 17, 2025

Slow compilation when taking elements of 2D array

bugSomething isn't workingSomething isn't working

Status: Open.

#32704 In jax-ml/jax;

· SUSYUSTC opened on Oct 16, 2025

Serializing an exported function fails when parameterized by a dictionary with integer keys

bugSomething isn't workingSomething isn't working

Status: Open.

#32680 In jax-ml/jax;

· mdenil opened on Oct 16, 2025

Add support for <code>mean</code> parameter in <code>std</code> and `var

contributions welcomeThe JAX team has not prioritized work on this. Community contributions are welcome.The JAX team has not prioritized work on this. Community contributions are welcome.

enhancementNew feature or requestNew feature or request

P2 (eventual)This ought to be addressed, but has no schedule at the moment. (Assignee optional)This ought to be addressed, but has no schedule at the moment. (Assignee optional)

Status: Open.

Feature

#32648 In jax-ml/jax;

· jakevdp opened on Oct 15, 2025