sympy2jax 将SymPy表达式转换为参数化，可微分，可矢量化的JAX函数。 所有SymPy浮点数都将成为可训练的输入参数。 SymPy符号成为传递矩阵的列。 安装 pip install git+https://github.com/MilesCranmer/sympy2jax.git 例子 import sympy from sympy import symbols import jax import jax . numpy as jnp from jax import rando