flaxOptimizersBenchmark:对标代码以评估亚麻优化器

亚麻优化基准 该存储库的目标是为用户提供一组评估Flax优化器的基准。 我们的目标是： 给出一些快速基准以快速评估新的优化器 提供经典的基准测试和绘图功能，以帮助希望发布新的优化程序的深度学习论文的作者 安装 您可以使用以下命令安装此librarie： pip install git+https://github.com/nestordemeure/flaxOptimizersBenchmark.git 用法 快速基准 去做 基准测试慢 去做 亚麻优化器 包含Flax中大量优化器的实现。 包含我在亚麻中对Adahessian二阶优化器的实现。 包含官方的Flax优化器。