MiniGPT with JAX and 3D Parallelism — Continued

Part 2: Multinode training with Jax and Ray