equinox
equinox copied to clipboard
FYI: JAX 0.7.0 is known to be buggy with Equinox.
Pinning an issue to highlight that JAX 0.7.0 is known to cause segfaults when used with Equinox. Right now JAX 0.6.* is recommended instead.
Reference: https://github.com/jax-ml/jax/issues/30517
FYI I'll just note here that as of JAX 0.7.0, python >=3.11 is required. This wouldn't be related to the issue but is important to know.