Alternatives and similar repositories for jax-flash-attn3

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

• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆21Updated 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
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆25Updated last year
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆19Updated this week
• LaurentMazare / glim
  ☆20Updated 8 months ago
• huggingface / candle-paged-attention
  ☆12Updated last year
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆24Updated 3 months ago
• UmerHA / triton_util
  Make triton easier
  ☆46Updated last year
• LaurentMazare / gemm-metal
  ☆15Updated 7 months ago
• BBuf / flash-rwkv
  ☆31Updated last year
• EricLBuehler / safetensors_explorer
  CLI utility to inspect and explore .safetensors and .gguf files
  ☆20Updated 2 weeks ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 11 months ago
• megvii-research / IntLLaMA
  IntLLaMA: A fast and light quantization solution for LLaMA
  ☆18Updated last year
• hazan-lab / flash-stu
  PyTorch implementation of the Flash Spectral Transform Unit.
  ☆17Updated 9 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…
  ☆23Updated this week
• huggingface / kernel-builder
  👷 Build compute kernels
  ☆68Updated this week
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆59Updated last year
• WaveSpeedAI / QuantumAttention
  [WIP] Better (FP8) attention for Hopper
  ☆30Updated 4 months ago
• belindal / state-tracking
  Code and data for paper "(How) do Language Models Track State?"
  ☆14Updated 2 months ago
• YdrMaster / cuda-driver
  基于 CUDA Driver API 的 cuda 运行时环境
  ☆16Updated last week
• gordicaleksa / OpenGemini
  Effort to open-source 10.5 trillion parameter Gemini model.
  ☆17Updated 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 …
  ☆16Updated last year
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆61Updated 2 months ago
• EricLBuehler / candle_graphs
  Graph model execution API for Candle
  ☆13Updated 7 months ago
• sgl-project / DeepGEMM
  DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
  ☆17Updated 2 weeks ago
• RWKV / ZeroCoT
  https://x.com/BlinkDL_AI/status/1884768989743882276
  ☆28Updated last month
• herrmann / rustorch
  "PyTorch in Rust"
  ☆16Updated last year
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated 2 years ago
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆23Updated 5 months ago