文件名称：sympy2jax：将SymPy表达式转换为JAX函数
文件大小：10KB
文件格式：ZIP
更新时间：2024-03-23 18:22:13
Python
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 random from sympy2jax import sympy2jax 让我们在SymPy中创建一个表达式： x , y = symbols ( 'x y' ) expression = 1.0 * sympy . cos ( x ) + 3.2 * y 让我们获取JAX版本。 我们传递方程式和所
【文件预览】：
sympy2jax-master
----sympy2jax()
--------__init__.py(56B)
--------sympy2jax.py(5KB)
----.github()
--------workflows()
----LICENSE(11KB)
----setup.py(2KB)
----README.md(2KB)
----tests()
--------test_some.py(1KB)
----.gitignore(38B)