PySR / pysr /export_jax.py

Commit History

Fixed jax export compatibility with refactor
73c6ffd

tttc3 commited on

Add mappings from sympy exact One/Half
f2a7a62

MilesCranmer commited on

Add export key error telling user to set function mappings
99fff5c

MilesCranmer commited on

Propagate and check torch/jax mappings
e0c68fc

MilesCranmer commited on

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

deepsource-autofix[bot] commited on

Format code with black
7d4300a
unverified

deepsource-autofix[bot] commited on

Fix sympy2jax for selection of indices
55e3b83

MilesCranmer commited on

Allow custom selection of X matrix in torch/jax modules
8c55475

MilesCranmer commited on

Refactor lazy loading of torch and jax
fb950bb

MilesCranmer commited on

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

MilesCranmer commited on

Move JAX export to separate file
5b978f9

MilesCranmer commited on