利用 PyTorch 加速非线性最小化
天体物理仪器与方法
2018-11-20 v1 数据分析、统计与概率
摘要
我展示了主要面向神经网络训练的软件框架 PyTorch 可轻易应用于科学中的一般函数最小化问题。PyTorch 易用与极高效率的特质在此领域同样适用,并以极小的软件工程代价带来求解时间上两个数量级的改进。
引用
@article{arxiv.1805.07439,
title = {Acceleration of Non-Linear Minimisation with PyTorch},
author = {Bojan Nikolic},
journal= {arXiv preprint arXiv:1805.07439},
year = {2018}
}
备注
5 pages, 2 figures, 2 program listings. Full source code available at: https://github.com/bnikolic/oof/