Search results

Inconsistent results from <code>jax.random.categorical</code> with <code>bfloat16</code> vs <code>float32</code> probabilities

bugSomething isn't workingSomething isn't working

Status: Open.

#30879 In jax-ml/jax;

· spktrm opened on Aug 10, 2025

Conda Installation with Cuda

Status: Open.

#30874 In jax-ml/jax;

· TristynAlxander opened on Aug 9, 2025

TRACING CACHE MISS ... explanation unavailable! please open an issue at https://github.com/jax-ml/jax

bugSomething isn't workingSomething isn't working

Status: Open.

#30871 In jax-ml/jax;

· geligeli opened on Aug 9, 2025

Locally include static elements in a PyTree traversal?

enhancementNew feature or requestNew feature or request

Status: Open.

#30860 In jax-ml/jax;

· adonath opened on Aug 8, 2025

Hardcoded backends for donation argument support

bugSomething isn't workingSomething isn't working

Status: Open.

#30852 In jax-ml/jax;

· shelkesagar29 opened on Aug 8, 2025

<code>lax.reduce</code> NotImplementedError: Reduction computations can't close over Tracers

bugSomething isn't workingSomething isn't working

Status: Open.

#30841 In jax-ml/jax;

· jeertmans opened on Aug 8, 2025

Failed to initialize with TPU v3

bugSomething isn't workingSomething isn't working

Status: Open.

#30834 In jax-ml/jax;

· neverix opened on Aug 8, 2025

Greedy evaluation of <code>custom_jvp</code> fails when invoked in <code>jax.ensure_compile_time_eval()</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#30787 In jax-ml/jax;

· otoomey opened on Aug 6, 2025

5090 error

bugSomething isn't workingSomething isn't working

Status: Open.

#30786 In jax-ml/jax;

· sunbin1993 opened on Aug 6, 2025

Corruption error messages when script completes execution

bugSomething isn't workingSomething isn't working

Status: Open.

#30731 In jax-ml/jax;

· jusevitch opened on Aug 4, 2025

jax is very very slower very slow in gpu than cpu

bugSomething isn't workingSomething isn't working

Status: Open.

#30708 In jax-ml/jax;

· ak24watch opened on Aug 2, 2025

division of the result of jnp.mean(are, where=mask) by itself produces nan under @jax.jit

bugSomething isn't workingSomething isn't working

Status: Open.

#30707 In jax-ml/jax;

· kurolesica0211 opened on Aug 2, 2025