Commits

Breadcrumbs

History for

onpod

User selector

All users

All time

Commit History

Commits on Mar 14, 2022

• fix style

  borisdayma

  committed
  39e0862

  Copy full SHA for 39e0862

Commits on Mar 9, 2022

• style: reformat

  borisdayma

  committed
  fc526dc

  Copy full SHA for fc526dc

• feat: check args

  borisdayma

  committed
  17cd7d4

  Copy full SHA for 17cd7d4

• feat(type): handle grafting types

  borisdayma

  committed
  602da65

  Copy full SHA for 602da65

• feat: update distributed_shampoo

  borisdayma

  committed
  933a89f

  Copy full SHA for 933a89f

Commits on Mar 1, 2022

• feat: update demo

  borisdayma

  committed
  02225fb

  Copy full SHA for 02225fb

• feat: update demo

  borisdayma

  committed
  94b0a3e

  Copy full SHA for 94b0a3e

Commits on Feb 28, 2022

• feat: add blank_caption_prob

  borisdayma

  committed
  e668e0b

  Copy full SHA for e668e0b

Commits on Feb 27, 2022

• feat: auto start_preconditioning_step

  borisdayma

  committed
  4efa049

  Copy full SHA for 4efa049

• doc: batch size per dp device

  borisdayma

  committed
  9bed35d

  Copy full SHA for 9bed35d

• feat: update shampoo matrix epsilon

  borisdayma

  committed
  1ad4ee5

  Copy full SHA for 1ad4ee5

• feat: vmap trick is the new default

  borisdayma

  committed
  fd1a3e7

  Copy full SHA for fd1a3e7

Commits on Feb 26, 2022

• feat: vmap trick for eval

  borisdayma

  committed
  57edfbe

  Copy full SHA for 57edfbe

• feat: try vmap trick

  borisdayma

  committed
  889658c

  Copy full SHA for 889658c

• feat: no dropout

  borisdayma

  committed
  cae6369

  Copy full SHA for cae6369

• feat: no vmap trick on pod

  borisdayma

  committed
  efd6a19

  Copy full SHA for efd6a19

• doc: fix comment

  borisdayma

  committed
  62e0539

  Copy full SHA for 62e0539

• feat: add grad_param_spec

  borisdayma

  committed
  72a3ecb

  Copy full SHA for 72a3ecb

• debug: no vmap trick

  borisdayma

  committed
  d027eb8

  Copy full SHA for d027eb8

• feat: add grad spec

  borisdayma

  committed
  fca1a2a

  Copy full SHA for fca1a2a

• feat: sharding constraint on cumul

  borisdayma

  committed
  edd0ed6

  Copy full SHA for edd0ed6

• feat: cleanup

  borisdayma

  committed
  57dfd33

  Copy full SHA for 57dfd33

• feat: add lots of sharding constraints

  borisdayma

  committed
  a784354

  Copy full SHA for a784354

• debug: load on cpu

  borisdayma

  committed
  a39f39f

  Copy full SHA for a39f39f

• feat: enforce no vmap

  borisdayma

  committed
  f0ac7bb

  Copy full SHA for f0ac7bb

• feat: update vmap trick

  borisdayma

  committed
  ff4e76d

  Copy full SHA for ff4e76d

• fix: 2D parallel

  borisdayma

  committed
  b42396a

  Copy full SHA for b42396a

Commits on Feb 25, 2022

• feat: no vmap trick on multi-node

  borisdayma

  committed
  ba170e0

  Copy full SHA for ba170e0

• feat: multi_host -> use_vmap_trick

  borisdayma

  committed
  6deb0a4

  Copy full SHA for 6deb0a4

• refactor: cleanup partition spec

  borisdayma

  committed
  c9deeca

  Copy full SHA for c9deeca

• fix: local dp device + constraint on vmap trick

  borisdayma

  committed
  6d267e1

  Copy full SHA for 6d267e1

• feat: regular pjit for eval_loss

  borisdayma

  committed
  fc504c7

  Copy full SHA for fc504c7

• refactor: reshape batch in training loop

  borisdayma

  committed
  ab5e0f3

  Copy full SHA for ab5e0f3

• refactor(train): cleanup

  borisdayma

  committed
  ae265a1

  Copy full SHA for ae265a1

• feat: update configs

  borisdayma

  committed
  b8095e9

  Copy full SHA for b8095e9