google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
What it solves
JAXopt 提供了一个优化器库，其具有硬件加速、可批处理、可微分的特性。它允许用户在 GPU 和 TPU 上解决优化问题，并针对其输入对这些问题的解进行微分。
How it works
该库基于 JAX，利用 JAX 的 vmap 进行自动向量化，以处理同时处理多个相同的优化问题实例。它支持通过隐式微分或通过对展开的算法迭代使用 autodiff 来对优化解进行微分。
Who it's for
需要能够集成到可微分编程工作流中的高性能优化工具，且正在使用 JAX 的研究人员和开发者。
Highlights\n
- CPU、GPU 和 TPU 的硬件加速。
- 通过 JAX 的
vmap进行自动向量化，以批处理优化问题。 - 支持优化解的隐式和展开式微分。
相关
- 项目
- 项目
- 项目
- 项目