mup
Maximal Update Parametrization (μP) 提供了一种适用于大规模神经网络的稳定超参数优化方法,例如预训练的Transformer等。利用μP,模型的超参数在不同大小的网络中可保持稳定,减少了在探索和扩展过程中的不确定性和脆弱性。该工具包简化了在PyTorch模型中实现μP的流程,是优化和调优深度学习模型的强大工具。