开源项目

按分类、标签筛选，或直接搜索项目名称。

分类标签排序

共 2 个项目 · 标签：JAX

xAI 开源的 Grok-1 大模型权重与 JAX 示例代码，314B 参数 MoE 架构，支持加载与推理测试。

Python 52k 8.5k

LLM AI MoE 开源模型 JAX

Keras 3 是一个多后端深度学习框架，支持 JAX、TensorFlow、PyTorch 和 OpenVINO，让开发者用同一套高级 API 构建和训练模型，兼顾开发效率与运行性能。

Python 64k 2k

Python AI 深度学习 PyTorch TensorFlow JAX