Skip to content
@NX-AI

NXAI

Pinned Loading

  1. xlstm Public

    Official repository of the xLSTM.

    Python 1.8k 139

  2. xlstm-jax Public

    Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.

    Python 91 2

  3. mlstm_kernels Public

    Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.

    Jupyter Notebook 54 2

Repositories

Showing 9 of 9 repositories
  • transformers Public Forked from huggingface/transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    Python 1 Apache-2.0 29,171 0 0 Updated Apr 20, 2025
  • xlstm Public

    Official repository of the xLSTM.

    Python 1,829 Apache-2.0 139 35 3 Updated Apr 7, 2025
  • mlstm_kernels Public

    Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.

    Jupyter Notebook 54 2 3 0 Updated Apr 6, 2025
  • flashrnn Public

    FlashRNN - Fast RNN Kernels with I/O Awareness

    Python 82 4 0 1 Updated Mar 26, 2025
  • RULER Public Forked from NVIDIA/RULER

    This repo contains the source code for RULER: What’s the Real Context Size of Your Long-Context Language Models?

    Python 0 Apache-2.0 81 0 0 Updated Feb 4, 2025
  • xlstm-jax Public

    Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.

    Python 91 2 1 0 Updated Jan 8, 2025
  • vllm Public Forked from vllm-project/vllm

    A high-throughput and memory-efficient inference and serving engine for LLMs

    Python 0 Apache-2.0 7,073 0 0 Updated Jan 2, 2025
  • NeuralDEM Public

    Real-time Simulation of Industrial Particulate Flows

    14 0 0 0 Updated Nov 15, 2024
  • vision-lstm Public

    xLSTM as Generic Vision Backbone

    Python 472 Apache-2.0 34 1 0 Updated Nov 4, 2024

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Loading…

Most used topics

Loading…