文章图片

如果你是一个没有数学或软件背景但想学习深度学习的初学者 ， 那么你不会想使用 JAX 。 相反 ， Keras 是更好的选择 。
不该使用 JAX 的四条理由
虽然上文已经讨论了很多 JAX 的正面反馈 ， 它有潜力极大地提升用户程序的性能 。 但作者同时列举了以下四条不该使用 JAX 的理由：

• JAX 仍然被官方认为是一个实验性框架 。 JAX 是一个相对「年轻」的项目 。 目前 ， JAX 仍被视为一个研究项目 ， 而不是成熟的谷歌产品 ， 因此如果用户正在考虑迁移到 JAX ， 请记住这一点；
• 使用 JAX 一定要勤勉 。 调试的时间成本 ， 或者更严重的是 ， 未跟踪副作用（untracked side effects）的风险可能导致那些没有扎实掌握函数式编程的用户不适用 JAX 。 在开始将它用于正式项目之前 ， 请确保自己了解使用 JAX 的常见缺陷；
• JAX 没有针对 CPU 计算进行优化 。 鉴于 JAX 是以「加速器优先」的方式开发的 ， 因此每个操作的分派并未针对 JAX 进行完全优化 。 在某些情况下 ， NumPy 实际上可能比 JAX 更快 ， 尤其是对于小型程序而言 ， 这是因为 JAX 引入了开销；
• JAX 与 Windows 不兼容 。 目前在 Windows 上不支持 JAX 。 如果用户使用 Windows 系统但仍想尝试 JAX ， 可以使用 Colab 或将其安装在虚拟机（VM）上 。

原文链接：https://www.assemblyai.com/blog/why-you-should-or-shouldnt-be-using-jax-in-2022/