Code

Add checkpointed_scan#60

Code

Open

dlwh wants to merge 8 commits intomainstanford-crfm/haliax:mainfrom ckpt_scanstanford-crfm/haliax:ckpt_scanCopy head branch name to clipboard

+251-20

Commits

Commits on Jan 21, 2024

• log the name that's being scanned

  dlwh

  committed
  bf37fdb

• parens seem more normal

  dlwh

  committed
  962dff2

• change default prevent_cse to False for Stacked

  dlwh

  committed
  f668bd8

• support fancier checkpointed_scan in Stacked

  dlwh

  committed
  689df5a

• oops

  dlwh

  committed
  eef2a0e

• more logging

  dlwh

  committed
  ee7b32b

• more logging!

  dlwh

  committed
  ea47079

• what

  dlwh

  committed
  b91ecfc