看啥推荐读物
专栏名称: 爱可可-爱生活
知名互联网资讯博主 北邮PRIS模式识别实验室陈老师
今天看啥  ›  专栏  ›  爱可可-爱生活

【Mu-Transformer:使用 Jax/Flax 实现的 -20240506135345

爱可可-爱生活  · 微博  · AI  · 2024-05-06 13:53
2024-05-06 13:53 本条微博链接 【Mu-Transformer:使用 Jax/Flax 实现的 Transformer 模型,支持 Mu-Parameterization,能在 TPU pods 上运行 FSDP】'Mu-Transformer - Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.' GitHub: github.com/lucaslingle/mu_transformer #开源# #机器学习# #人工智能# ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照