热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#JAXopt
jaxopt - JAX驱动的高性能优化器库
JAXopt
优化器
硬件加速
批处理
可微分
Github
开源项目
JAXopt是基于JAX的优化器库,提供硬件加速、批处理和可微分的优化算法。支持GPU和TPU,可自动向量化多个优化问题实例,并支持优化解的微分。目前正与Optax合并,处于维护模式。适用于机器学习和科学计算领域的优化任务。
1
1
相关文章
JAXopt: 硬件加速、可批处理和可微分的JAX优化器
3 个月前
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号