【文件属性】：
文件名称：sympy2jax：将SymPy表达式转换为JAX函数
文件大小：10KB
文件格式：ZIP
更新时间：2021-03-04 00:35:33
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)