6 ms·
> with torch.compile the main advantage of jax is disappearing. Interesting take - I agree here somewhat. But also, wouldn't you think a framework that has be
by n7g 2y ago
> with torch.compile the main advantage of jax is disappearing.
Interesting take - I agree here somewhat.
But also, wouldn't you think a framework that has been from the ground-up designed around a specific, mature compiler stack be better able to integrate compilers in a more stable fashion than just shoe-horning static compilers into a very dynamic framework? ;)
- wafngar 2y agoDepends. 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..
- anon389r58r58 2y agoSo the answer is not Jax? Because JAX is not designed around a mature compiler stack. The history of Jax is more so that it matured alongside the compiler...