EiffL / jax-gpu-cluster-demo

Little demo of how to instantiate a cluster of GPUs with JAX

Geek Repo

Github PK Tool