Alternatives and similar repositories for jax-flash-attn3

Users that are interested in jax-flash-attn3 are comparing it to the libraries listed below

• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last year
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆19Updated this week
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆26Updated last year
• BBuf / flash-rwkv
  ☆32Updated last year
• UmerHA / triton_util
  Make triton easier
  ☆48Updated last year
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆29Updated this week
• frankxwang / dpo-prefix-sharing
  DPO, but faster 🚀
  ☆45Updated 10 months ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated last year
• kyegomez / MobileVLM
  Implementation of the LDP module block in PyTorch and Zeta from the paper: "MobileVLM: A Fast, Strong and Open Vision Language Assistant …
  ☆14Updated last year
• nanowell / Q-Sparse-LLM
  My Implementation of Q-Sparse: All Large Language Models can be Fully Sparsely-Activated
  ☆33Updated last year
• foundation-model-stack / bamba
  Train, tune, and infer Bamba model
  ☆134Updated 4 months ago
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆73Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆122Updated last month
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated 2 years ago
• tridao / flash-attention-wheels
  ☆57Updated last year
• dusty-nv / NanoDB
  Zero-copy multimodal vector DB with CUDA and CLIP/SigLIP
  ☆61Updated 5 months ago
• RWKV / ZeroCoT
  https://x.com/BlinkDL_AI/status/1884768989743882276
  ☆28Updated 5 months ago
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆25Updated 7 months ago
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆25Updated last week
• microsoft / AttentionEngine
  ☆101Updated 5 months ago
• graphcore-research / jax-scalify
  JAX Scalify: end-to-end scaled arithmetics
  ☆16Updated 11 months ago
• opendatahub-io / vllm-tgis-adapter
  vLLM adapter for a TGIS-compatible gRPC server.
  ☆41Updated this week
• chenwanqq / candle-llava
  implement llava using candle
  ☆15Updated last year
• casper-hansen / AutoAWQ_kernels
  ☆78Updated 11 months ago
• yixiaoer / tpu-training-example
  ☆15Updated last year
• michaelfeil / candle-flash-attn-v3
  ☆12Updated 8 months ago
• NX-AI / mlstm_kernels
  Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.
  ☆72Updated this week
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆27Updated last week
• onnx / neural-compressor
  Model compression for ONNX
  ☆97Updated 11 months ago
• IST-DASLab / QIGen
  Repository for CPU Kernel Generation for LLM Inference
  ☆26Updated 2 years ago