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
  ☆13Updated 5 months ago
• tile-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆18Updated this week
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated last year
• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆21Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated 6 months ago
• drbh / nnli
  🔭 interactively explore `onnx` networks in your CLI.
  ☆23Updated 10 months ago
• herrmann / rustorch
  "PyTorch in Rust"
  ☆16Updated last year
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆13Updated 9 months ago
• kyutai-labs / kaudio
  Rust crate for some audio utilities
  ☆22Updated last month
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• NVIDIA / free-threaded-python
  No-GIL Python environment featuring NVIDIA Deep Learning libraries.
  ☆57Updated last week
• LaurentMazare / glim
  ☆19Updated 6 months ago
• huggingface / candle-paged-attention
  ☆12Updated last year
• UmerHA / triton_util
  Make triton easier
  ☆47Updated 10 months 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 …
  ☆16Updated last year
• TiledTensor / TiledBench
  Benchmark tests supporting the TiledCUDA library.
  ☆16Updated 5 months ago
• chenwanqq / candle-llava
  implement llava using candle
  ☆14Updated 10 months ago
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆38Updated 2 months ago
• LaurentMazare / tch-ext
  Sample Python extension using Rust/PyO3/tch to interact with PyTorch
  ☆34Updated last year
• Zyphra / zcookbook
  Training hybrid models for dummies.
  ☆20Updated 3 months ago
• sdpython / onnxcustom
  Tutorial on how to convert machine learned models into ONNX
  ☆16Updated 2 years ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆18Updated 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
• zaydzuhri / pythia-mlkv
  Multi-Layer Key-Value sharing experiments on Pythia models
  ☆32Updated 10 months ago
• Ryu1845 / hyena-jax
  Implementation of Hyena Hierarchy in JAX
  ☆10Updated last year
• Deep-Learning-Profiling-Tools / triton-samples
  ☆13Updated last month
• caibucai22 / awesome-cuda
  Awesome code, projects, books, etc. related to CUDA
  ☆16Updated last week
• Yuanhy1997 / HyPe
  HyPe: Better Pre-trained Language Model Fine-tuning with Hidden Representation Perturbation [ACL 2023]
  ☆14Updated last year
• latentCall145 / channels-last-groupnorm
  A CUDA kernel for NHWC GroupNorm for PyTorch
  ☆18Updated 5 months ago
• dusty-nv / NanoDB
  Zero-copy multimodal vector DB with CUDA and CLIP/SigLIP
  ☆54Updated 10 months ago