Multi-GPUs training with JAX and equinox
Training neural networks with multiple GPUs using JAX and equinox
2024-02-29