Repository Issue Activity (beta)

jax-ml/jax

Current issue state, recent activity, and per-issue timelines from the indexed issue data.

Open Issues
1,591
New in 7 Days
11
Closed in 7 Days
5
Average Open Age
806 days
Stale 30+ Days
1,503
Stale 90+ Days
1,371
Last 2 Weeks
DateOpenedClosedCommentsEventsOpen Backlog
2026-08-2400000
2026-08-2310135
2026-08-2220221
2026-08-2110043
2026-08-2023493
2026-08-1911113
2026-08-1811234
2026-08-1730113
2026-08-1610113
2026-08-1531166
2026-08-1450256
2026-08-1331254
2026-08-12014144
2026-08-115513346
This Week

Opened: 8

Closed: 5

Comments: 10

Events: 22

Top Labels
bug (3421)
enhancement (1323)
question (348)
NVIDIA GPU (238)
documentation (198)
pallas (141)
needs info (132)
P2 (eventual) (104)
Issue Explorer
IssueAuthorStateLabelsCommentsReactionsUpdated

#39449 `lax.cond` creates unnecessary copies when one branch is a noop

Opened 29 days ago
Gattocrucco
open
bug
201 day ago

#39621 [Pallas TPU] Support SparseCore-to-TensorCore Remote DMA on Cloud TPU

Opened 24 days ago
ayaka14732
open - reopened
enhancement
441 day ago

#40143 GPU: multi-dimensional `irfft` lowering (since 0.10.0) is ~1.7× slower than the single cuFFT C2R op for Hermitian-symmetric inputs — request an opt-in for the direct path

Opened 1 day ago
hbahk
open
enhancement
001 day ago

#40142 Repeated XLA compiles of jnp.argsort on the CUDA backend crashes

Opened 2 days ago
quantum-ml-geek
open
bug
002 days ago

#40141 `eigh` forward- and reverse-mode eigenvector gradients return all-`NaN` for eigenvalues above `2^p` (mantissa-precision cliff in the JVP rule's div-by-zero guard)

Opened 2 days ago
shadow-orange41
open
bug
002 days ago

#40128 splash attention: `dsinks` gradient loses precision by rounding before the reduction

Opened 3 days ago
ds-hwang
open
bug
202 days ago

#40110 `jax_explain_cache_misses` causes custom_vmap to crash

Opened 4 days ago
skye-ant
closed - completed
bug
104 days ago

#39840 [Pallas:TPU:SparseCore] addupdate_scatter crashes InferVectorLayoutPass with layout passes enabled

Opened 15 days ago
samixyzdev
open
No labels
204 days ago

#39899 `jnp.linalg.norm` narrows float16 back *before* the `sqrt`, discarding the float32 headroom it just paid for (wrong primal, silently-zero gradient)

Opened 13 days ago
shadow-orange41
open
bug
204 days ago

#32098 Clarify FFI call semantics in presence of sharding

Opened 11 months ago
jjyyxx
closed - completed
enhancement
304 days ago

#40101 XLA:CPU: a small dynamic-update-slice inside a loop rewrites the whole buffer each iteration (0.11.1 regression)

Opened 4 days ago
mmcky
open
No labels
304 days ago

#32165 Warnings on multi-process launch

Opened 11 months ago
dest1n1s
open
bug
304 days ago

#40093 [Pallas TPU] Support TPU v4 BarnaCore

Opened 5 days ago
ayaka14732
open
enhancement
104 days ago

#36958 Unexpected result involving `scan` and `vmap`

Opened 4 months ago
puct9
closed - completed
bug
404 days ago

#35358 Nondeterministic Error "couldn't get temp CUBIN file name"

Opened 6 months ago
ajshawn
closed - completed
bug
525 days ago

#39905 `jnp.linalg.det` is numerically unstable for 2x2 and 3x3 inputs: wrong sign / large relative error, and spurious `inf`

Opened 13 days ago
shadow-orange41
closed - completed
bug
105 days ago

#40081 Mosaic GPU `broadcast_in_dim` unnecessarily restricts ranks > 2

Opened 6 days ago
samanklesaria
open
bug
006 days ago

#40020 jnp.histogram on an empty array raises ValueError; NumPy returns zero counts with default (0, 1) edges

Opened 9 days ago
shashvat-singham
closed - completed
No labels
006 days ago

#40048 Support out-of-bounds predication for the `cp.async` for GMEM→SMEM (pre-Hopper)

Opened 7 days ago
samanklesaria
open
enhancement
006 days ago

#18440 scatter_add in float32 may be inaccurate when accumulating large numbers of entries

Opened 3 years ago
jakevdp
open
bug
306 days ago

#36927 Performance regression in batched jnp.linalg.solve on CPU from jax/jaxlib 0.9.2 to 0.10.0

Opened 4 months ago
clemisch
open
bug
706 days ago

#40038 [RFC] Unsafe opaque PJRT C-API handles on jax.Array for external PJRT-C-API interop

Opened 7 days ago
lokic233
open
No labels
007 days ago

#40028 jnp.remainder gives a zero result the sign of the dividend, not the divisor

Opened 7 days ago
krrishapatel
open
No labels
107 days ago

#40026 Allow dicts with mixed-type keys by comparing by (type name, value)

Opened 8 days ago
carlosgmartin
open
enhancement
008 days ago

#39998 `splat_is_compatible_with_tiled` left-aligns the shapes

Opened 10 days ago
samanklesaria
open
bug
109 days ago

Rows per page:

1–25 of 6,129