Jax is a competing computational framework that does something similar to PyTorch, so both of your questions don't really make sense.
its like fretting about how everything should be written in C++ instead of Python/Javascript
I don't care.
if not im not interested. i'll keep using pytorch.
i hope this answer makes more sense for you.