pymc-devs/pytensor

解决的问题

PyTensor 提供了一种定义、优化和评估包含多维数组的复杂数学表达式的方法。它作为 PyMC 的计算后端，实现了高效的数值计算和梯度计算。

工作原理

它使用符号图框架来表示数学运算。与一些现代深度学习库不同，PyTensor 维护一个静态图，该图可以就地修改以实现高级优化。它可以通过 C、JAX 或 Numba 将这些图转换为可执行代码，以确保高性能。

适用人群

专为需要灵活、可修改的符号数学计算后端的开发者和研究人员设计，特别是从事概率编程或自定义符号优化的人群。

主要亮点

• 静态图优化：允许就地修改表达式图，以移除不必要的操作。
• 多后端编译：支持通过 C、JAX 和 Numba 进行编译。
• 自动微分：能够计算符号表达式的梯度。
• 可扩展框架：纯 Python 代码库，支持快速开发自定义操作符。