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
[WIP] begin implementation of multi-output compatibility b5b74c3 MilesCranmer commited on May 28, 2021