jax.example_libraries.optimizers.adam(step_size, b1=0.9, b2=0.999, eps=1e-08)
构建 Adam 的优化器三元组。...jax.example_libraries.optimizers.adamax(step_size, b1=0.9, b2=0.999, eps=1e-08)
为 AdaMax(基于无穷范数的 Adam...=1.0)
jax.example_libraries.optimizers.rmsprop(step_size, gamma=0.9, eps=1e-08)
为 RMSProp 构造优化器三元组。...jax.example_libraries.optimizers.rmsprop_momentum(step_size, gamma=0.9, eps=1e-08, momentum=0.9)
为带动量的...使用 call() 调用 TensorFlow 函数,支持反向模式自动微分
主机计算的另一个可能用途是调用为另一个框架编写的库,如 TensorFlow。