#jax

GPT-J-6B：基于6B的基于JAX的变压器
GPT-J-6B: 6B JAX-Based Transformer(arankomatsuzaki.wordpress.com)
2021-6-9 21:4
GPT-J-6B在各种零射流下游任务上与6.7B GPT-3（或yie）进行几乎按照6.7b gpt-3（或yie）。 GPT-J是在各种下游任务上的零射击性能方面最正常的公共变压器LM。 该项目需要大量的人数 - 小时时间比其他大规模模型开发所做的，这表明JAX + XMAP + TPU是用于快速开发大型模型的......
JAX与Tensorflow Vs. pytorch与朱莉娅
Jax vs. TensorFlow vs. Pytorch vs. Julia(www.reddit.com)
2021-3-13 4:33
我想知道有人是否可以比较雅克和朱莉娅，以及两者的利益和缺陷。 为什么Google创建JAX而不是...
使用JAX加速研究
Using JAX to Accelerate Research(deepmind.com)
2020-12-18 20:30
DeepMind工程师通过构建工具，扩展算法并创建具有挑战性的虚拟和物理世界来训练和测试人工智能（AI）系统，从而加快了我们的研究速度。作为这项工作的一部分，我们不断评估新的机器学习库和框架。 最近，我们发现Google研究团队开发的机器学习框架JAX为越来越多的项目提供了很好的服务。 JAX与我们的工程理念非常吻......
使用JAX加速研究
Using JAX to Accelerate Research(deepmind.com)
2020-12-5 8:5
DeepMind工程师通过构建工具，扩展算法并创建具有挑战性的虚拟和物理世界来训练和测试人工智能（AI）系统，从而加快了我们的研究速度。作为这项工作的一部分，我们不断评估新的机器学习库和框架。 最近，我们发现Google研究团队开发的机器学习框架JAX为越来越多的项目提供了很好的服务。 JAX在我们的工程哲学中引起......