google-deepmind / dks

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.
68Updated last week

Alternatives and similar repositories for dks:

Users that are interested in dks are comparing it to the libraries listed below