No idea how to fix this. I've tried to install caustics on an M1 Mac dozens of times and succeeded only once with some specific permutation of compiler, Python and JAX versions. It might just be easier to rewrite the root solver entirely in JAX than to fix this.