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: add mini_glu config
a7e5050

boris commited on

feat: update mini config
d9a16f2

boris commited on

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

boris commited on

feat: support pod (#139)
803ccbf
unverified

boris commited on

feat: update configs
79557f9

boris commited on

feat: add micro config
e501f71

boris commited on

feat: create config files
dc5c024

boris commited on

feat: add config
3f13951

boris commited on