艾久久

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

  • 爱可可-爱生活
  • 2024-12-18 21:44:09
【NX-AI mLSTM Kernels：提供了一套快速高效的 mLSTM 核心库，支持并行、分块和循环步骤的处理方式，并提供了 PyTorch 和 JAX 的封装】'NX-AI/mlstm_kernels: A library for fast and efficient mLSTM Kernels.' GitHub: github.com/NX-AI/mlstm_kernels 深度学习 mLSTM AI库 #AI创造营#
NX-AI mLSTM Kernels：提供了一套快速高效的 mLSTM 核心库，支持并行、分块和循环步骤的处理方式，并提供了 PyTorch 和 JAX 的封装