dm haiku:基于JAX的神经网络库 源码

Haiku:for | | | | 什么是Hai句? ku句是一个工具用于构建神经网络想一想:“for ” Haiku是由的一些作者开发的的简单神经网络库, 是的神经网络库。 消除歧义:如果您正在寻找Haiku操作系统,请参阅 。 注意:Haiku目前是beta版。 许多研究人员对Haiku进行了几个月的测试,并大规模复制了许多实验。 请随时使用Haiku,如果您有任何问题,请! 总览 是一个数值计算库,结合了NumPy,自动微分和一流的GPU / TPU支持。 Haiku是用于JAX的简单神经网络库,它使用户可以使用熟悉的面向对象的编程模型,同时允许完全访问JAX的纯函数转换