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
  ☆26Updated 5 months ago
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated last year
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆22Updated 3 weeks 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
  ☆20Updated last year
• BBuf / flash-rwkv
  ☆30Updated 10 months ago
• Doraemonzzz / Awesome-Triton-Resources
  Awesome Triton Resources
  ☆23Updated this week
• BBuf / tensorrt-llm-moe
  ☆23Updated last month
• huggingface / candle-paged-attention
  ☆12Updated last year
• LaurentMazare / gemm-metal
  ☆12Updated 4 months ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆53Updated 3 weeks 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 last month
• kyegomez / OpenStrawberry
  An open source replication of the stawberry method that leverages Monte Carlo Search with PPO and or DPO
  ☆28Updated 3 weeks ago
• 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
• RWKV / ZeroCoT
  https://x.com/BlinkDL_AI/status/1884768989743882276
  ☆27Updated last month
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆35Updated last month
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆60Updated 11 months ago
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆23Updated 9 months ago
• recursal / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆45Updated 8 months ago
• LaurentMazare / glim
  ☆19Updated 5 months ago
• TiledTensor / TiledBench
  Benchmark tests supporting the TiledCUDA library.
  ☆15Updated 4 months ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 8 months ago
• SmerkyG / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆10Updated 2 weeks ago
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆20Updated 2 months ago
• yixiaoer / tpu-training-example
  ☆14Updated 8 months ago
• facebookresearch / coocmap
  code for paper "Accessing higher dimensions for unsupervised word translation"
  ☆21Updated last year
• facebookresearch / adaptive_scheduling
  Experimental scripts for researching data adaptive learning rate scheduling.
  ☆23Updated last year
• kyutai-labs / moshi-webrtc
  Proof of concept for running moshi/hibiki using webrtc
  ☆18Updated last month
• WaveSpeedAI / QuantumAttention
  [WIP] Better (FP8) attention for Hopper
  ☆26Updated last month
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆42Updated 6 months ago