This enable x64 corrupts JAX on my laptop
https://github.com/cosmodesi/cosmoprimo/blob/f3698636242fe54587fe3a607f4b14acb8efc9ac/cosmoprimo/fftlog.py#L19C35-L19C35
This is surely the fault of the flaky JAX on my laptop, but it seems you could turn it off without affecting others by wrapping in something like
if jax.default_backend() != "METAL" :
... and cosmoprimo still works at least in some cases... not sure where expected to cause problems...
This enable x64 corrupts JAX on my laptop
https://github.com/cosmodesi/cosmoprimo/blob/f3698636242fe54587fe3a607f4b14acb8efc9ac/cosmoprimo/fftlog.py#L19C35-L19C35
This is surely the fault of the flaky JAX on my laptop, but it seems you could turn it off without affecting others by wrapping in something like
if jax.default_backend() != "METAL" :
... and cosmoprimo still works at least in some cases... not sure where expected to cause problems...