使用 JAX 进行深度学习
2024-10-01
这篇文章主要介绍了 Manning 出版社即将出版的一本新书《Deep Learning with JAX》。这本书介绍了如何使用 Google 的高性能数值计算库 JAX 来加速深度学习和其他计算密集型任务。该书涵盖了 JAX 的核心概念、如何使用 JAX 构建可微模型、执行分布式计算以及利用 JAX 生态系统中的库和模块等内容,适合有一定 Python 基础和深度学习知识的读者阅读。
19
未分类
数值计算