diffrax icon indicating copy to clipboard operation
diffrax copied to clipboard
verified publisher icon patrick-kidger

FYI: JAX 0.7.0 and 0.7.1 are known to be buggy with Diffrax.

Open patrick-kidger opened this issue 4 months ago • 0 comments

Pinning an issue to highlight that when used with Diffrax:

  • JAX 0.7.0 is known to cause segfaults (https://github.com/jax-ml/jax/issues/30517)
  • JAX 0.7.1 is known to cause major performance issues (https://github.com/jax-ml/jax/issues/31284). Right now JAX 0.6.* is recommended instead.

patrick-kidger avatar Aug 21 '25 10:08 patrick-kidger