Depends. PyTorch on the other hand has a large user base and well defined and tested api. So should be doable; and is already progressing and rapid speed..
PyTorch is developed by multiple companies / stake holders while jax is google only with internal tooling they don’t share with the world. This alone is a major reason not to use jax. Also I think it is more the other way around: with torch.compile the main advantage of jax is disappearing.
US has no bureaucrats? All the EU member states, no bureaucrats? How many projects in member states cost the tax payer billions? Btw the EU budget is tiny in comparison.
The UK would do much better had they not chosen Brexit. Basically everyone here admits that (bank of England etc) apart from some zealot politicians. And the health system has collapsed. The problems are not solely caused by Brexit but it exacerbates nearly all of them.