Commit History

feat(train): improve pjit speed
f254058

boris commited on

fix(train): consider correct batch size
b7c7458

boris commited on

style: unsused import
7a176b9

boris commited on

feat(train): progress on pjit
49597a2

boris commited on

feat(data): support accumulation in non-streaming
88c8e06

boris commited on

feat: custom gradient accumulation
2d07559

boris commited on

fix(data): no shuffling of validation data
ddcbc6a

boris commited on

fix(train): handle seed_dataset
8b72ed8

boris commited on

feat: support pypi
f5dba1e

boris commited on