Commit History

  1. feat: vmap optimizer (#166)

    Unverified signature
    b993d27

  1. feat: scan layers + gradient checkpointing (#161)⁠

    Unverified signature
    07a6f9a

  1. feat: better multi-node support (#158)⁠

    Unverified signature
    728a3c3

  1. feat: implement transformer variants (#144)⁠

    Unverified signature
    542378c

  1. feat: support pod (#139)

    Unverified signature
    803ccbf

  1. feat: create config files

    dc5c024

  1. feat: add config

    3f13951