Update jit helpers to allow for polymorphic coeffects