NX-AI mLSTM Kernels：提供了一套快速高效的 mLSTM 核心库，支持并行、分块和循环步骤的处理方式，并提供了 PyTorch 和 JAX 的封装

【NX-AI mLSTM Kernels：提供了一套快速高效的 mLSTM 核心库，支持并行、分块和循环步骤的处理方式，并提供了 PyTorch 和 JAX 的封装】

2024-12-18浏览详情