EagerPy：编写可原生兼容 PyTorch、TensorFlow、JAX 与 NumPy 的代码
机器学习
2020-08-11 v1 数学软件
机器学习
摘要
EagerPy 是一个 Python 框架，可让你编写自动原生兼容 PyTorch、TensorFlow、JAX 和 NumPy 的代码。库开发者不再需要在仅支持其中一个框架，或为各框架重实现库并处理代码重复之间做选择。此类库的用户可更轻松地切换框架，而不被某个特定的第三方库锁定。除多框架支持外，EagerPy 还为任何框架带来全面的类型标注以及对方法链的一致支持。最新文档可在线查阅 https://eagerpy.jonasrauber.de，代码可在 GitHub https://github.com/jonasrauber/eagerpy 获取。
引用
@article{arxiv.2008.04175,
title = {EagerPy: Writing Code That Works Natively with PyTorch, TensorFlow, JAX, and NumPy},
author = {Jonas Rauber and Matthias Bethge and Wieland Brendel},
journal= {arXiv preprint arXiv:2008.04175},
year = {2020}
}