Search results

Profiling Traces Disappear

bugSomething isn't workingSomething isn't working

Status: Open.

#32276 In jax-ml/jax;

· cdelv opened on Oct 2, 2025

vmap should work on ref-operating functions (i.e. sub-kernels) inside a Pallas TPU context

bugSomething isn't workingSomething isn't working

pallasIssues pertaining to Pallas (GPU or TPU)Issues pertaining to Pallas (GPU or TPU)

Status: Open.

#32272 In jax-ml/jax;

· summivox opened on Oct 1, 2025

Incorrect handling of explicit padding in lax.conv_transpose

bugSomething isn't workingSomething isn't working

Status: Open.

#32267 In jax-ml/jax;

· samanklesaria opened on Oct 1, 2025

cuDNN dot_product_attention NaNs with FP32 bias

bugSomething isn't workingSomething isn't working

Status: Open.

#32242 In jax-ml/jax;

· sbodenstein opened on Oct 1, 2025

jnp.histogram on uint4 array behaves different than other dtypes

bugSomething isn't workingSomething isn't working

Status: Open.

#32231 In jax-ml/jax;

· adityaiitb opened on Oct 1, 2025

Implement scipy.ndimage.gaussian_filter

enhancementNew feature or requestNew feature or request

Status: Open.

#32225 In jax-ml/jax;

· chlofisher opened on Sep 30, 2025

<code>ragged_dot</code> silently ignores <code>DotAlgorithmPreset</code> precisions

bugSomething isn't workingSomething isn't working

Status: Open.

#32207 In jax-ml/jax;

· chr1sj0nes opened on Sep 30, 2025

out of memory when used multi env setting (vmap)

bugSomething isn't workingSomething isn't working

Status: Open.

#32201 In jax-ml/jax;

· ak24watch opened on Sep 30, 2025

Parallel Outer product proposition

enhancementNew feature or requestNew feature or request

Status: Open.

#32194 In jax-ml/jax;

· DiagRisker opened on Sep 29, 2025

MakeFromNodeDataAndChildren: how to replace?

enhancementNew feature or requestNew feature or request

Status: Open.

#32186 In jax-ml/jax;

· PhilipVinc opened on Sep 29, 2025

🐛 Bug Report: GPU matmul fails with INTERNAL: an unsupported value or parameter was passed to the function

bugSomething isn't workingSomething isn't working

Status: Open.

#32168 In jax-ml/jax;

· Ascarshen opened on Sep 29, 2025

building jax 0.7.2 from source fails on Jetson Orin Nano

bugSomething isn't workingSomething isn't working

Status: Open.

#32166 In jax-ml/jax;

· notnot opened on Sep 28, 2025