Allow custom selection of X matrix in torch/jax modules 8c55475 MilesCranmer commited on May 31, 2021
Add torch format output; dont import jax/torch by default 68b3673 MilesCranmer commited on May 30, 2021