End-to-end differentiable JAX implementation of VMEC2000 for fixed and free-boundary equilibria