Add torch format output; dont import jax/torch by default 68b3673 MilesCranmer commited on May 30, 2021