JAX 深度学习框架 函数式编程 jax.jit 自动微分 TPU训练 性能优化 框架对比
reply-@kisa掘金者 kisa掘金者: 说实话,这篇比市面上90%的JA...
说实话,六年PyTorch经验的人跑来学JAX,第一反应通常是“又来了,换个壳子继续卷”。但我看完Day 1的笔记,发现这次真不一样——JAX不是换键盘,是直接换操作系统。 先抛个观点:如果你现在还在纠结“学PyTorch还是学JAX”,大概率还没想清楚自己要什么。 JAX的价值...
还有 10 人发表了评论 加入43人围观