静态缓存页面 · 查看动态版本 · 登录

GoMLX 项目近况

✨步子哥 · 2025-09-24T06:21:11+00:00

截至 2025 年 8 月，GoMLX 仍处于“早期可用”阶段：核心训练-推理链路已跑通，但距离“生产级”还有明显缺口。 1. 功能完成度 • 基础算子：全连接、CNN、LSTM、Multi-Head Attention、KAN、FFT 等已合入主干；ResNet-18/34 可在单 GPU 上完整训练，但 ResNet-50 尚缺 SeparableConv2D，官方放在 TODO 里。 • 优化器：SGD / Adam / AdamW 已可用，LAMB、Shampoo 等还在 roadmap。 • 自动微分与即时编译：依托 OpenXLA/PJRT，CPU、CUDA 后端都能跑，训练 32-128 样本的小 batch 速度与 PyTorch-CPU 基本持平。 • 预训练模型：暂无直接加载 HuggingFace 权重的能力；社区给出了 onnx-gomlx 小工具，可先把 PyTorch 模型转 ONNX，再导进 GoMLX 做推理。 2. 近期进展（2024H2-2025H1） • v0.3 → v0.4 主要把 CUDA kernel 调用开销降了 30

✨

✨步子哥 @steper · 2025-10-01 14:17

这个项目可以实现在浏览器环境运行，有一点的潜力。如果Go的后续版本支持 wasm 3.0的话，潜力很大。

暂无表态