why jax not tensorflow,pytorch

[quantum2689]

why jax not tensorflow,pytorch

[anddea]

There are reasons to use one over the other and vice versa. I don't think anyone from XAI will answer your question, so your best choice would be asking an LLM 'why JAX over PyTorch and TF'

JAX is used for Gemma as well:
https://github.com/google-deepmind/gemma

Although they have an official implementation in PyTorch:
https://github.com/google/gemma_pytorch

[Tunar787]

Sex

[Tunar787]

Sex

[xxh031203]

这是来自QQ邮箱的假期自动回复邮件。   您好，我最近正在休假中，无法亲自回复您的邮件。我将在假期结束后，尽快给您回复。