Commit History

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

    Unverified signature
    728a3c3

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

    Unverified signature
    542378c

  1. feat(data): super conditioning (#141)⁠

    Unverified signature
    7939874

  1. feat: support pod (#139)

    Unverified signature
    803ccbf

  1. feat(train) - handle multiple nodes (#130)

    Unverified signature
    0952927
  2. feat: handle model parallel

    1bb3269

  1. feat(train): improve pjit speed

    f254058
  2. fix(train): consider correct batch size

    b7c7458

  1. style: unsused import

    7a176b9

  1. feat(train): progress on pjit

    49597a2

  1. feat(data): support accumulation in non-streaming

    88c8e06
  2. feat: custom gradient accumulation

    2d07559

  1. fix(data): no shuffling of validation data

    ddcbc6a
  2. fix(train): handle seed_dataset

    8b72ed8

  1. feat: support pypi

    f5dba1e