A more accurate title for the OP would be "I hope and wish PyTorch were dead, so Jax could become the standard." Leaving aside the fact that PyTorch's ecosyste

A more accurate title for the OP would be "I hope and wish PyTorch were dead, so Jax could become the standard." Leaving aside the fact that PyTorch's ecosystem is 10x to 100x larger, depending on how one measures it, PyTorch's biggest advantage, in my experience, is that it can be picked up quickly by developers who are new to it. Jax, despite its superiority, or maybe because of it, can not be picked up quickly. Equinox does a great job of making Jax accessible, but Jax's functional approach is in practice more difficult to learn than PyTorch's object-oriented one.