google/sympy2jax

Turn SymPy expressions into trainable JAX expressions.

Pythonsympyjax
This is stars and forks stats for /google/sympy2jax repository. As of 06 May, 2024 this repository has 272 stars and 12 forks.

sympy2jax Turn SymPy expressions into trainable JAX expressions. The output will be an Equinox module with all SymPy floats (integers, rationals, ...) as leaves. SymPy symbols will be inputs. Optimise your symbolic expressions via gradient descent! Installation pip install sympy2jax Requires: Python 3.7+ JAX 0.3.4+ Equinox 0.5.3+ SymPy 1.7.1+. Example import jax import sympy import sympy2jax x_sym = sympy.symbols("x_sym") cosx = 1.0 * sympy.cos(x_sym) sinx = 2.0 * sympy.sin(x_sym) mod = sympy2jax.SymbolicModule([cosx,...
Read on GithubGithub Stats Page
repotechsstarsweeklyforksweekly
winterknife/PINKPANTHERAssemblyPython4920580
dvf/blockchainC#PythonJavaScript7.5k02.8k0
tefkah/zotero-nightSCSSTypeScriptPython1.7k+17300
ClementTsang/bottomRustPython7.6k+76191+1
sharkdp/hyperfineRustPython17.6k+92299+2
rgcr/m-cliShellPython9.5k03120
UVVM/UVVMVHDLHTMLPython3050790
Shougo/deol.nvimVim ScriptTypeScriptPython3240160
banteg/yearn-feesVyperPython30010
Tinkerforge/warp-chargerHTMLJavaScriptTeX45070