jax

abe4f613 - revive old numpy promotion logic