JAX - 资源平台：AI工具、网站、APP和开源项目合集
GradCache
Gradient Cache对比学习
突破GPU/TPU内存限制，实现对比学习无限扩展
dm_pix
PIXJAX
基于JAX的高性能图像处理库
penzai
PenzaiJAX
用于构建、编辑和可视化神经网络的 JAX 研究工具包
EasyLM
EasyLMJAX
简化的大规模语言模型训练与部署
keras-nlp
KerasNLPTensorFlow
兼容多框架的自然语言处理工具和预训练模型
EasyDeL
EasyDeL机器学习
多模型训练优化框架
keras
Keras 3深度学习框架
多后端支持的深度学习框架，兼容JAX、TensorFlow和PyTorch
dopamine
Dopamine强化学习
用于快速原型设计的强化学习研究框架
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号