Allow custom selection of X matrix in torch/jax modules 8c55475 MilesCranmer commited on May 31, 2021
Merge pull request #48 from MilesCranmer/torch-export 45b290b unverified MilesCranmer commited on May 31, 2021
Add torch format output; dont import jax/torch by default 68b3673 MilesCranmer commited on May 30, 2021