Spaces:

MilesCranmer
/

PySR

Sleeping

App Files Files Community

PySR / pysr /export_jax.py

Commit History

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