PySR / pysr /export_jax.py

Commit History

test: fix mypy errors with sympy
e84bed4
unverified

MilesCranmer commited on

Merge pull request #650 from MilesCranmer/deps-update
1327c58
unverified

MilesCranmer commited on

fix: use of numpy in jax expressions
723dbce
unverified

MilesCranmer commited on

fix: symbolic numbers in jax
144e3ff
unverified

MilesCranmer commited on

Add pre-commit hooks for unused imports
b933dea
unverified

MilesCranmer commited on

Fix sympy2jax for rational numbers
17c9b1a

MilesCranmer 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