前言本文适用对象：任何接触过 TensorFlow, Pytorch, Keras 并且已经开始了解或尝鲜 Jax 的人群。如果是没有接触过任何深度学习框架的人群，这篇文章可能不适合你。在开始学习之前，你应该对 PyTorch 或 TensorFlow 有一定的了解。Jax 可能是一个比较难学的库，但值得一学。为什么使用 Jax 的理由这里就不多赘述了。就我个人而言，tf 或者 torch 自定义
Copyright © 2005-2024 51CTO.COM 版权所有 京ICP证060544号
