jax

dbe4bdf9 - add multibackend option to soft_pmap