wls2002 69d73aab73 add backend="jax" to sympy module

2024-06-13 05:55:33 +08:00

..

__init__.py

add sympy support; which can transfer your network into sympy expression;

2024-06-12 21:36:35 +08:00

act_jnp.py

add sympy support; which can transfer your network into sympy expression;

2024-06-12 21:36:35 +08:00

act_sympy.py

add backend="jax" to sympy module

2024-06-13 05:55:33 +08:00