注册会员 | 设为首页 | 加入收藏夹
您好，欢迎光临本网站！[请登录] ！[注册会员]！  

搜索资源列表

  1. mesh-transformer-jax:在Jax和Haiku中为并联变压器建模-源码

  2. 网格变压器Jax 一个xmap库，使用Jax中新的（有文献证明） xmap运算符来实现变压器的模型并行性。 有关使用此示例实现自回归语言模型的示例，请参见enwik8_example.py 。 去做 解散头和碎片（当前头==碎片） TPU上的测试/基准 实现梯度检查点 分片激活，而不是复制

  3. 所属分类：其它

    • 发布日期：2021-03-14
    • 文件大小：4096
    • 提供者：weixin_42112658