Maybe deisteve will answer for himself, but I don't think that's meant to mean how to run Jax on Pytorch, but rather the questions he's interested in for any published model.
its like fretting about how everything should be written in C++ instead of Python/Javascript
I don't care.