PySR / pysr /export_torch.py

Commit History

Switch to using sympytorch
b0e1209

MilesCranmer commited on

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

MilesCranmer commited on

Remove unused torch->sympy code
b80fb14

MilesCranmer commited on

Allow user to pass extra torch operators
e7ede78

MilesCranmer commited on

Remove unused torch backwards to sympy
a88a169

MilesCranmer commited on

Add docstring for sympy2torch
9068541

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

Create torch export function
a06bfc4

MilesCranmer commited on