jax

1d652ab7 - Don't recompute source_info for each tracer during staging.