Inconsistent environment variable names

[gabeweisz]

MaxText uses the environment variables JAX_COORDINATOR_IP, JAX_COORDINATOR_PORT, NNODES, and NODE_RANK for multi-system GPU training, but JAX_COORDINATOR_ADDRESS, a fixed port, JAX_PROCESS_COUNT, and a combination of several environment variables, and for multi-system CPU training. It would be great if both configurations used the same environment variables

[shralex]

Since this changes the "api", perhaps we could introduce duplicate variables, and mark others as deprecated while still respecting them for a while.