jax

c7829385 - use cleaner process_primitive implementation