[D] Deepspeed vs JAX for distributed training

Are there benchmarks that show speedups/resource utilization between distributed training with JAX ecosystem and deepspeed? preferably on GPUs for fair analysis, from my understanding JAX/FLAX can squeeze resources from TPU pods but I think deepsped can't? (might be wrong)

analysis benchmarks deepspeed distributed ecosystem fair gpus jax machinelearning resources show think tpu training understanding