文件名称：Haiku：基于JAX的神经网络库-python
文件大小：561KB
文件格式：ZIP
更新时间：2024-07-08 14:14:37
机器学习
Haiku：基于JAX的神经网络库，允许用户使用熟悉的面向对象程序设计模型，可完全访问 JAX 的纯函数变换 俳句：JAX 的十四行诗概览 | 为什么是俳句？ | 快速入门 | 安装 | 示例 | 用户手册 | 文档 | 引用俳句 什么是俳句？ Haiku 是一个简单的 JAX 神经网络库，由 Sonnet（TensorFlow 的神经网络库）的一些作者开发。 注意：Haiku 目前是 alpha 版。 许多研究人员已经对 Haiku 进行了几个月的测试，并大规模复制了许多实验。 请随意使用 Haiku，但一定要测试任何假设，如果事情看起来不对，请告诉我们！ 概述 JAX 是一个数值计算库，结合了 NumPy、自动微分和一流的 GPU/TPU 支持。 Haiku 是一个简单的 JAX 神经网络库，它使用户能够使用熟悉的面向对象编程模型，同时允许完全访问 JAX 的纯函数转换。 Haiku 提供了两个核心工具：模块抽象 hk.Module 和简单的函数转换 hk.transform。 hk.Modules 是 Python 对象，它们保存对它们自己的参数、