yzhwang / jax-multi-gpu-resnet50-exampleLinks
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
Sorting: