Optax关键词检索结果

2021年3月19日 00:00

从头开始​​在 JAX 中构建 Transformer：如何编写和训练自己的模型

Build a Transformer in JAX from scratch: how to write and train your own models

如何使用 JAX、Haiku 和 Optax 开发和训练 Transformer。通过示例学习如何在 JAX 中编写深度学习模型

查看详细页面