jax_scalify
