马斯克说到做到开源Grok-1，开源社区一片狂喜。

但基于Grok-1做改动or商用，都还有点难题：

Grok-1使用Rust+JAX构建，对于习惯Python+PyTorch+HuggingFace等主流软件生态的用户上手门槛高。

△图注：Grok登上GitHub热度榜世界第一

Colossal-AI团队最新成果，解大家燃眉之急，提供方便易用的Python+PyTorch+HuggingFace Grok-1，能将推理时延加速近4倍！

现在，模型已在HuggingFace、ModelScope上发布。

HuggingFace下载链接：
https://huggingface.co/hpcai-tech/grok-1

ModelScope下载链接：
https://www.modelscope.cn/models/colossalai/grok-1-pytorch/summary

性能优化

结合Colossal-AI在AI大模型系统优化领域的丰富积累，已迅速支持对Grok-1的张量并行。

在单台8H800 80GB服务器上，推理性能相比JAX、HuggingFace的auto device map等方法，推理时延加速近4倍。

使用教程

下载安装Colossal-AI后，启动推理脚本即可。

./run_inference_fast.sh hpcaitech/grok-1