Commit History

feat: vmap optimizer (#166)
b993d27
unverified

boris commited on

feat: scan layers + gradient checkpointing (#161)
07a6f9a
unverified

boris commited on

feat: better multi-node support (#158)
728a3c3
unverified

boris commited on

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

boris commited on

feat: support pod (#139)
803ccbf
unverified

boris commited on

feat: create config files
dc5c024

boris commited on

feat: add config
3f13951

boris commited on