Skip to main content

Browse the archive

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
Raw File
setup.cfg 
[flake8]
max-line-length = 88
ignore =
  C901 # object names too complex
  E111, E114 # four-space indents
  E121 # line continuations
  W503, W504 # line breaks around binary operators
max-complexity = 18
select = B,C,F,W,T4,B9,E225,E227,E228
exclude =
  .git,
  build,
  __pycache__
per-file-ignores =
  docs/autodidax.py:F811
back to top