yzhwang / jax-multi-gpu-resnet50-example

An example showing how to use jax to train resnet50 on multi-node multi-GPU
20Updated 2 years ago

Alternatives and similar repositories for jax-multi-gpu-resnet50-example:

Users that are interested in jax-multi-gpu-resnet50-example are comparing it to the libraries listed below