Code

lax.full: add sharding argument#19445

Code

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom jakevdp:full-devicejakevdp/jax:full-deviceCopy head branch name to clipboardJan 22, 2024

+17-2

Commits

Commits on Jan 22, 2024

• lax.full: add sharding argument

  jakevdp

  committed
  f41a32c

  Copy full SHA for f41a32c