ekpass
Results
1
comments of
ekpass
Based on the traceback you get when you set JAX_TRACEBACK_FILTERING=off, it looks like the problem ultimately stems from the maybe_jvp_tracer function of jax/_src/interpreters/ad.py. So a hacky solution that allowed the...