Jax AI工具大全：热门网站、应用及开源项目推荐
purejaxrl
PureJaxRL强化学习
JAX强化学习框架实现千倍性能提升
ssm-book
状态空间模型机器学习
状态空间模型可执行教程，配合dynamax库的实践指南
ringattention
Ring AttentionBlockwise Transformers
创新注意力机制大幅提升Transformer上下文处理能力
big_vision
big vision深度学习
基于Jax/Flax的大规模视觉模型训练框架
prompt-tuning
Prompt TuningT5X
简介提示微调技术的参数优化及其实践方法
nanodl
NanoDLJax
设计与训练变压器模型的Jax库
maxtext
MaxTextTPU
高性能与可扩展的开源大模型解决方案，支持TPUs与GPUs
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号