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 / gemm-metal
  ☆14Updated 5 months ago
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆18Updated last week
• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆21Updated last year
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated last year
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 9 months ago
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆24Updated 11 months ago
• LaurentMazare / glim
  ☆19Updated 7 months ago
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆23Updated 2 months ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆60Updated last month
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 7 months ago
• huggingface / candle-paged-attention
  ☆12Updated last year
• belindal / state-tracking
  ☆14Updated last month
• BBuf / flash-rwkv
  ☆30Updated 11 months ago
• huggingface / candle-cublaslt
  ☆13Updated last year
• sgl-project / DeepGEMM
  DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling
  ☆15Updated this week
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆38Updated last year
• caibucai22 / awesome-cuda
  Awesome code, projects, books, etc. related to CUDA
  ☆16Updated last month
• megvii-research / IntLLaMA
  IntLLaMA: A fast and light quantization solution for LLaMA
  ☆18Updated last year
• li-plus / flash-preference
  Accelerate LLM preference tuning via prefix sharing with a single line of code
  ☆41Updated 2 weeks ago
• TiledTensor / TiledBench
  Benchmark tests supporting the TiledCUDA library.
  ☆16Updated 5 months ago
• SusCom-Lab / ZSMerge
  ☆16Updated 3 weeks ago
• MILVLG / mlc-imp
  Enable everyone to develop, optimize and deploy AI models natively on everyone's devices.
  ☆10Updated 11 months ago
• BlinkDL / LM-Trick-Questions
  Here we collect trick questions and failed tasks for open source LLMs to improve them.
  ☆32Updated 2 years ago
• chenwanqq / candle-llava
  implement llava using candle
  ☆14Updated 11 months ago
• latentCall145 / channels-last-groupnorm
  A CUDA kernel for NHWC GroupNorm for PyTorch
  ☆18Updated 6 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
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆39Updated 3 months ago
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• EricLBuehler / candle_graphs
  Graph model execution API for Candle
  ☆14Updated 5 months ago
• ptillet / triton-llvm-releases
  ☆21Updated last year