issues
search
HomebrewNLP
/
Olmax
HomebrewNLP in JAX flavour for maintable TPU-Training
BSD 2-Clause "Simplified" License
45
stars
5
forks
source link
Improve Shampoo
#97
Closed
ClashLuke
closed
1 year ago
ClashLuke
commented
1 year ago
Allow FP64
Flatten Conv + Depth (both as flags)
Re-add shampoo for all layers (previously only input + output)