Search results

<code>jaxlib</code> 0.5.3 causes different autodiff behavior for equivalent functions

bugSomething isn't workingSomething isn't working

Status: Open.

#31600 In jax-ml/jax;

· tetterl opened on Sep 5, 2025

FFI calls on GPU lead to segmentation fault

bugSomething isn't workingSomething isn't working

Status: Open.

#31593 In jax-ml/jax;

· RezaNajian opened on Sep 5, 2025

conv_general_dilated assertion on reverse mode autodiff

bugSomething isn't workingSomething isn't working

type:Bug

Status: Open.

#31592 In jax-ml/jax;

· manuelgloeckler opened on Sep 5, 2025

Out-of-bounds read in jnp.nonzero under checkify

bugSomething isn't workingSomething isn't working

Status: Open.

#31568 In jax-ml/jax;

· garylinscott opened on Sep 4, 2025

When mesh is set globally via <code>jax.set_mesh</code>, applying some jax transformations on arrays located on a single device raise a <code>ValueError</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#31458 In jax-ml/jax;

· acecchini opened on Sep 1, 2025

Test failures on darwin since 0.7.0

bugSomething isn't workingSomething isn't working

Status: Open.

#31428 In jax-ml/jax;

· GaetanLepage opened on Aug 29, 2025

Edit lax.convert_element_type to emit deprecation warning on overflow

enhancementNew feature or requestNew feature or request

Status: Open.

#31426 In jax-ml/jax;

· carlosgmartin opened on Aug 28, 2025

PJRT GPU crash for ppermute in collective memory space.

bugSomething isn't workingSomething isn't working

Status: Open.

#31398 In jax-ml/jax;

· yliu120 opened on Aug 28, 2025

Incorrect indexing with jnp.int8 and Python scalar

bugSomething isn't workingSomething isn't working

Status: Open.

#31396 In jax-ml/jax;

· dodgebc opened on Aug 28, 2025

Show documentation for previous versions

enhancementNew feature or requestNew feature or request

Status: Open.

#31380 In jax-ml/jax;

· PAUL-BERNARD opened on Aug 27, 2025

<code>fftn</code> on small arrays in function called via <code>jax.lax.scan</code> crashes with missing external function <code>___xla_cpu_runtime_LegacyDuccFft</code>

bugSomething isn't workingSomething isn't working

Status: Open.

#31374 In jax-ml/jax;

· SJSeckmeyer opened on Aug 27, 2025

jax.lax.linalg.eigh performance cliff for larger batch sizes (on GPUs)

NVIDIA GPUIssues specific to NVIDIA GPUsIssues specific to NVIDIA GPUs

performancemake things lean and fastmake things lean and fast

Status: Open.

#31368 In jax-ml/jax;

· jaro-sevcik opened on Aug 27, 2025