蚁工厂 26-01-02 10:50
微博认证:科技博主

batson/mlux 是一个开源的 Python 库,该项目为 MLX 模型(运行在 Apple Silicon 芯片上的机器学习框架)提供可解释性分析工具。
github.com/batson/mlux

通常大模型像一个黑盒,你输入提示词,它输出文本。mlux 允许你深入观察模型每一层的内部状态。它的设计理念是让用户无需修改模型原本的推理(forward)或反向传播(backward)代码,即可插入“探针”或“钩子”(Hooks)。
灵感来源于 PyTorch 社区的 TransformerLens 和 JAX 社区的 Penzai,填补 MLX 生态中可解释性工具的空白。

#科技先锋官##AI创造营#

发布于 山东