Commit History

feat: implement transformer variants (#144)
542378c
unverified

boris commited on

feat(data): super conditioning (#141)
7939874
unverified

boris commited on

feat: support pod (#139)
803ccbf
unverified

boris commited on

feat(train) - handle multiple nodes (#130)
0952927
unverified

boris commited on

feat: handle model parallel
1bb3269

boris commited on

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