Solve JAX numerical computing tasks from a problem.json manifest. Handles reductions, vmap, grad, scan, and jit tasks.