https://github.com/google/jax
History
Tip revision: 34b3a7f106baa9099802f246fda6ad3476f32d26 authored by Qiao Zhang on 16 February 2022, 19:26:23 UTC
Allow mesh_util to map a logical axis to 3 physical axes, e.g. for pure data parallel training on TPU.
Tip revision: 34b3a7f
File Mode Size
jax_logo.png -rw-r--r-- 140.9 KB
jax_logo.svg -rw-r--r-- 3.9 KB
jax_logo_250px.png -rw-r--r-- 33.2 KB
jax_logo_500px.png -rw-r--r-- 48.1 KB
lifecycle.png -rw-r--r-- 74.5 KB

back to top