jax

b2c53a3e - [no-thunks] Reduce stack frames when tracing initial-style primitives.

[no-thunks] Reduce stack frames when tracing initial-style primitives. Along with 10a153e6ce, this reduces the stack depth due to `scan` from 11 frames to 5. Also adds a regression test for traceback depths.