flax:Flax是JAX的神经网络生态系统旨在提高灵活性 源码
亚麻:JAX的神经网络库和生态系统,旨在提高灵活性 | | 请参阅我们的,以了解有关亚麻的所有知识。 Flax最初是由Google Research的Brain团队中的工程师和研究人员(与JAX团队密切合作)创立的,现在与开源社区共同开发。 越来越多的Alphabet研究部门的数百人在使用Flax进行日常工作,同时。 Flax团队的任务是为Alphabet内以及更广泛的社区中不断发展的JAX神经网络研究生态系统提供服务,并探索JAX的应用案例。 我们使用GitHub进行几乎所有的协调和计划,以及讨论即将进行的设计更改的地方。 我们欢迎您对我们的任何讨论,问题和请求请求线程提供反馈。 我们正在将一些剩余的内部设计文档和对话线程移至GitHub讨论,问题和请求请求。 我们希望越来越多地参与更广泛的生态系统的需求和澄清。 请告诉我们怎么帮助你! 注意:新的亚麻现在稳定,我们建议所有
文件列表
flax:Flax是JAX的神经网络生态系统,旨在提高灵活性
(预估有个278文件)
.gitignore
109B
setup.cfg
74B
.gitignore
5B
Colab_Language_Model.ipynb
35KB
imagenet.ipynb
839KB
mnist.ipynb
106KB
pytest.ini
1KB
polyak-averaging.diff
3KB
ensembling.diff
4KB
checkpointing.diff
1KB
暂无评论