Revision - 34b3a7f - Allow mesh_util to map a logical axis to 3 physical axes, e.g. [...]

Revision 34b3a7f106baa9099802f246fda6ad3476f32d26 authored by Qiao Zhang on 16 February 2022, 19:26:23 UTC, committed by jax authors on 16 February 2022, 19:41:41 UTC

Allow mesh_util to map a logical axis to 3 physical axes, e.g. for pure data parallel training on TPU.

PiperOrigin-RevId: 429097675

1 parent e1fd630

Files
Changes

Permalinks

Lorentz_ODE_Solver.ipynb

Showing with 0 additions and 0 deletions (0 / 0 diffs computed)

Computing file changes ...