I haven't tracked it down yet, but the first commit in https://github.com/google/jax/pull/1256 fails with a core dump on OS X. I'm guessing this is due to some missing cleanup step when a Python exception is raised.