https://blog.overfit.org/tags/jax/flax/Recent content in JAX/Flax on 里列昂遗失的记事本Hugozh-cnFri, 15 Aug 2025 16:53:00 +0800https://blog.overfit.org/posts/2025-08-15-19039965-grain%E7%94%A8%E4%BA%8E%E8%AF%BB%E5%8F%96%E5%92%8C%E5%A4%84%E7%90%86%E7%94%A8%E4%BA%8E%E8%AE%AD%E7%BB%83%E5%92%8C%E8%AF%84%E4%BC%B0-jax-%E6%A8%A1%E5%9E%8B%E7%9A%84%E6%95%B0%E6%8D%AE/Fri, 15 Aug 2025 16:48:00 +0800https://blog.overfit.org/posts/2025-08-15-19039965-grain%E7%94%A8%E4%BA%8E%E8%AF%BB%E5%8F%96%E5%92%8C%E5%A4%84%E7%90%86%E7%94%A8%E4%BA%8E%E8%AE%AD%E7%BB%83%E5%92%8C%E8%AF%84%E4%BC%B0-jax-%E6%A8%A1%E5%9E%8B%E7%9A%84%E6%95%B0%E6%8D%AE/Grain介绍 Grain官方Docs 用户可以带来任意的 Python 转换。 Grain 被设计为模块化。如果需要，用户可以使用自己的实现轻松覆盖 Grain 组件。 同一管道的多次运行将产生相同的输出。 Grain 的设计使得检查点的大小最小。抢占后，Grain 可以从中断https://blog.overfit.org/posts/2025-08-15-19039904-lenetjaxflax%E5%AE%9E%E7%8E%B0/Fri, 15 Aug 2025 16:23:00 +0800https://blog.overfit.org/posts/2025-08-15-19039904-lenetjaxflax%E5%AE%9E%E7%8E%B0/LeNet LeNet，它是最早发布的卷积神经网络之一，因其在计算机视觉任务中的高效性能而受到广泛关注。 这个模型是由AT&T贝尔实验室的研究员Yann LeCun在1989年提出的（并以其命名），目的是识别图像 (LeCun et al., 1998)中的手写数字。 当