Alternatives and similar repositories for jax-flash-attn3

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

• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆24Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 8 months ago
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated last year
• LaurentMazare / gemm-metal
  ☆14Updated 6 months ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆60Updated last month
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆18Updated last week
• huggingface / candle-paged-attention
  ☆12Updated 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 …
  ☆15Updated last year
• habanero-lab / APPy
  APPy (Annotated Parallelism for Python) enables users to annotate loops and tensor expressions in Python with compiler directives akin to…
  ☆23Updated this week
• belindal / state-tracking
  Code and data for paper "(How) do Language Models Track State?"
  ☆14Updated 2 months ago
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 11 months ago
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆23Updated 2 months ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆39Updated 3 months ago
• huggingface / candle-cublaslt
  ☆13Updated last year
• huggingface / kernel-builder
  👷 Build compute kernels
  ☆44Updated this week
• LaurentMazare / glim
  ☆19Updated 8 months ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 10 months ago
• IST-DASLab / QIGen
  Repository for CPU Kernel Generation for LLM Inference
  ☆26Updated last year
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆21Updated 4 months ago
• sgl-project / DeepGEMM
  DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
  ☆15Updated 3 weeks ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆110Updated 8 months ago
• casper-hansen / AutoAWQ_kernels
  ☆74Updated 6 months ago
• SmerkyG / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆10Updated 3 weeks ago
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated last month
• BBuf / flash-rwkv
  ☆31Updated last year
• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆21Updated last year
• hazan-lab / flash-stu
  PyTorch implementation of the Flash Spectral Transform Unit.
  ☆17Updated 8 months ago
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• chenwanqq / candle-llava
  implement llava using candle
  ☆15Updated 11 months ago
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆29Updated last week