Search results

Inconsistent Reduction precision in backwards computation

bugSomething isn't workingSomething isn't working

Status: Open.

#30310 In jax-ml/jax;

· sbodenstein opened on Jul 18, 2025

Could you add the support of the new optimizer: Muon

enhancementNew feature or requestNew feature or request

Status: Open.

#30309 In jax-ml/jax;

· gzqccnu opened on Jul 18, 2025

<code>jax.tree_util.tree_map</code> fails when a registered pydantic object which has been copied using <code>deep=True</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#30299 In jax-ml/jax;

· jhewers-pf opened on Jul 18, 2025

XlaRuntimeError: INTERNAL: Failed to capture gpu graph: an unsupported value or parameter was passed to the function

bugSomething isn't workingSomething isn't working

Status: Open.

#30297 In jax-ml/jax;

· hwr9912 opened on Jul 18, 2025

jax.numpy.sort performance regression

bugSomething isn't workingSomething isn't working

Status: Open.

#30296 In jax-ml/jax;

· drewjenks01 opened on Jul 18, 2025

Json parse exceptions (and others) in perfetto traces

bugSomething isn't workingSomething isn't working

Status: Open.

#30290 In jax-ml/jax;

· dlwh opened on Jul 17, 2025

jax.numpy.fft.fftfreq no longer supports complex dtype

bugSomething isn't workingSomething isn't working

Status: Open.

#30287 In jax-ml/jax;

· kcdodd opened on Jul 17, 2025

[Feature Request] Add Sparse Attention kernel for GPUs in Pallas

enhancementNew feature or requestNew feature or request

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

Status: Open.

#30281 In jax-ml/jax;

· gmancusi opened on Jul 17, 2025

"ComputeCallSignature failed" in verbose logs investigating performance loss.

bugSomething isn't workingSomething isn't working

Status: Open.

#30270 In jax-ml/jax;

· cjchristopher opened on Jul 17, 2025

[Indexing] XlaRuntimeError: INVALID_ARGUMENT: Cannot concatenate arrays with different element types: S32 vs S64

bugSomething isn't workingSomething isn't working

Status: Open.

#30197 In jax-ml/jax;

· inailuig opened on Jul 14, 2025

Add custom derivative for scipy.special.hyp2f1

enhancementNew feature or requestNew feature or request

Status: Open.

#30195 In jax-ml/jax;

· mattbahr opened on Jul 14, 2025

jetson orin nx jetpack 6.2.1 cuSolver internal error

bugSomething isn't workingSomething isn't working

type:support

Status: Open.

#30188 In jax-ml/jax;

· hkortier opened on Jul 14, 2025