Spaces:

MilesCranmer
/

PySR

Running

App Files Files Community

PySR / pysr /export_jax.py

Commit History

Add mappings from sympy exact One/Half

f2a7a62

MilesCranmer commited on May 4, 2022

Add export key error telling user to set function mappings

99fff5c

MilesCranmer commited on Jun 11, 2021

Propagate and check torch/jax mappings

e0c68fc

MilesCranmer commited on Jun 11, 2021

Refactor unnecessary `else` / `elif` when `if` block has a `return` statement

5bb2875
unverified

deepsource-autofix[bot] commited on Jun 7, 2021

Format code with black

7d4300a
unverified

deepsource-autofix[bot] commited on Jun 7, 2021

Fix sympy2jax for selection of indices

55e3b83

MilesCranmer commited on May 31, 2021

Allow custom selection of X matrix in torch/jax modules

8c55475

MilesCranmer commited on May 31, 2021

Refactor lazy loading of torch and jax

fb950bb

MilesCranmer commited on May 30, 2021

Add torch format output; dont import jax/torch by default

68b3673

MilesCranmer commited on May 30, 2021

Move JAX export to separate file

5b978f9

MilesCranmer commited on May 30, 2021