ai-compiler-study / triton-kernels

Related projects ⓘ

Alternatives and complementary repositories for triton-kernels

• timudk / flux_triton
  Writing FLUX in Triton
  ☆30Updated last month
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last month
• cloneofsimo / imagenet.int8
  ☆33Updated 6 months ago
• aredden / torch-cublas-hgemm
  PyTorch half precision gemm lib w/ fused optional bias + optional relu/gelu
  ☆38Updated 2 months ago
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• mgmalek / efficient_cross_entropy
  ☆76Updated 5 months ago
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• huyphan168 / PEER
  Mixture of A Million Experts
  ☆31Updated 3 months ago
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 5 months ago
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• INT-FlashAttention2024 / INT-FlashAttention
  ☆46Updated last month
• cloneofsimo / efae
  ☆21Updated 4 months ago
• aredden / torch-bnb-fp4
  Faster Pytorch bitsandbytes 4bit fp4 nn.Linear ops
  ☆23Updated 7 months ago
• sekstini / basedxl
  ☆15Updated 7 months ago
• frankxwang / dpo-prefix-sharing
  DPO, but faster 🚀
  ☆21Updated 2 weeks ago
• brianfitzgerald / jax-mmdit
  Implementation of Diffusion Transformers and Rectified Flow in Jax
  ☆20Updated 4 months ago
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• cloneofsimo / repa-rf
  ☆26Updated last week
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• mapo-t2i / mapo
  Official codebase for Margin-aware Preference Optimization for Aligning Diffusion Models without Reference (MaPO).
  ☆60Updated 5 months ago
• BlinkDL / LinearAttentionArena
  Here we will test various linear attention designs.
  ☆56Updated 6 months ago
• surkovv / sdxl-unbox
  ☆28Updated last week
• SerChirag / rs-imle
  RS-IMLE
  ☆35Updated last month
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆40Updated last month
• xdit-project / mochi-xdit
  faster parallel inference of mochi video generation model
  ☆53Updated this week
• cloneofsimo / ezmup
  Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam
  ☆69Updated 3 months ago
• cloneofsimo / insightful-nn-papers
  These papers will provide unique insightful concepts that will broaden your perspective on neural networks and deep learning
  ☆45Updated last year
• philippe-eecs / small-vision
  A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.
  ☆29Updated 4 months ago
• davisyoshida / abnormal-floats
  Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)
  ☆18Updated last year