Search results

jax.scipy.optimize not accessible in 0.7.0

bugSomething isn't workingSomething isn't working

Status: Open.

#30491 In jax-ml/jax;

· kovnerd opened on Jul 25, 2025

Segmentation fault (core dumped) on H100

bugSomething isn't workingSomething isn't working

Status: Open.

#30481 In jax-ml/jax;

· rohandekate10 opened on Jul 24, 2025

Improve jax.config.update experience in interactive environments

enhancementNew feature or requestNew feature or request

Status: Open.

#30474 In jax-ml/jax;

· ryan112358 opened on Jul 24, 2025

Failed to determine best cudnn convolution algorithm for large kernel conv with large stride.

bugSomething isn't workingSomething isn't working

Status: Open.

#30469 In jax-ml/jax;

· Big-Baby opened on Jul 24, 2025

BCOO is not compatible with <code>jax.lax.scan</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#30466 In jax-ml/jax;

· inailuig opened on Jul 24, 2025

<code>visualize_array_sharding</code> unable to show device name on some shapes

bugSomething isn't workingSomething isn't working

Status: Open.

#30462 In jax-ml/jax;

· IvyZX opened on Jul 23, 2025

<code>jax.nn.standardize</code> returns <code>nan</code> when variance is lower than <code>-epsilon</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#30426 In jax-ml/jax;

· bondquant opened on Jul 23, 2025

Not Implemented functions in Pallas

bugSomething isn't workingSomething isn't working

Status: Open.

#30423 In jax-ml/jax;

· cdelv opened on Jul 23, 2025

Revisiting <code>custom_jvp</code> with <code>pure_callback</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#30401 In jax-ml/jax;

· bwohlberg opened on Jul 22, 2025

Pallas fails to write to output ref when using bfloat16 on TPU v3-8

bugSomething isn't workingSomething isn't working

Status: Open.

#30391 In jax-ml/jax;

· sr5434 opened on Jul 22, 2025

Profiling tool shows no GPU processes despite GPU Util being as expected.

bugSomething isn't workingSomething isn't working

Status: Open.

#30378 In jax-ml/jax;

· saschafrey opened on Jul 22, 2025

jet and equinox.nn.MLP

bugSomething isn't workingSomething isn't working

Status: Open.

#30352 In jax-ml/jax;

· A-Alaa opened on Jul 21, 2025