Closed joamatab closed 1 year ago
The alternative is to fix the current JAX release issues
I just released sax==0.10.3, which depends on klujax==0.2.3 where the issues mentioned in https://github.com/google/jax/issues/17662 were fixed.
Awesome, thank you Floris!
The alternative is to fix the current JAX release issues