shreyansh26 / Attention-Mask-PatternsView external linksLinks
Using FlexAttention to compute attention with different masking patterns
☆47Sep 22, 2024Updated last year
Alternatives and similar repositories for Attention-Mask-Patterns
Users that are interested in Attention-Mask-Patterns are comparing it to the libraries listed below
Sorting:
- ☆307Apr 23, 2025Updated 9 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- ☆17Aug 1, 2025Updated 6 months ago
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆54Jan 12, 2026Updated last month
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning☆21Nov 4, 2021Updated 4 years ago
- BFloat16 Fused Adam Operator for PyTorch☆16Nov 16, 2024Updated last year
- A PyTorch implementation of the paper https://arxiv.org/abs/1709.04875☆10Jul 22, 2020Updated 5 years ago
- Spectral Graph Attention Network with Fast Eigen-approximation☆12Dec 24, 2021Updated 4 years ago
- Energetic GraphNeural Networks (EGNN) implementation based on Dirichlet Energy Constrained Learning.☆27Nov 1, 2021Updated 4 years ago
- Fast and memory-efficient exact attention☆75Mar 3, 2025Updated 11 months ago
- ☆51Jan 28, 2024Updated 2 years ago
- Helpful tools and examples for working with flex-attention☆1,127Updated this week
- ☆15Oct 30, 2025Updated 3 months ago
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆13Dec 11, 2021Updated 4 years ago
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆235Jun 15, 2025Updated 7 months ago
- ☆124May 28, 2024Updated last year
- ☆32May 26, 2024Updated last year
- ☆31Jul 2, 2023Updated 2 years ago
- Pytorch (PyG) and Tensorflow (Keras/Spektral) implementation of Total Variation Graph Neural Network (TVGNN), as presented at ICML 2023.☆20Mar 15, 2025Updated 10 months ago
- ☆104Nov 7, 2024Updated last year
- Transformers components but in Triton☆34May 9, 2025Updated 9 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆167Aug 14, 2024Updated last year
- ☆16Oct 20, 2025Updated 3 months ago
- Memory-Based Meta-Learning on Non-Stationary Distributions☆17Mar 14, 2024Updated last year
- A basic pure pytorch implementation of flash attention☆16Oct 28, 2024Updated last year
- Efficient triton implementation of Native Sparse Attention.☆262May 23, 2025Updated 8 months ago
- [BMVC 2022] Information Theoretic Representation Distillation☆19Oct 6, 2023Updated 2 years ago
- Simple python library for generating your own perfetto traces for your application. Can be used for both app instrumentation and custom …☆24Jun 22, 2025Updated 7 months ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Jun 5, 2018Updated 7 years ago
- A token pruning method that accelerates ViTs for various tasks while maintaining high performance.☆28Jul 21, 2025Updated 6 months ago
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learning☆15Jun 10, 2024Updated last year
- Triton kernels for Flux☆22Jul 7, 2025Updated 7 months ago
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆152Mar 13, 2025Updated 11 months ago
- Homepage for ProLong (Princeton long-context language models) and paper "How to Train Long-Context Language Models (Effectively)"☆246Sep 12, 2025Updated 5 months ago
- A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.☆75Aug 2, 2024Updated last year
- Code for “ACE-HGNN: Adaptive Curvature ExplorationHyperbolic Graph Neural Network”☆17Mar 3, 2022Updated 3 years ago
- [SemEval'19] Code for "HLT@SUDA at SemEval 2019 Task 1: UCCA Graph Parsing as Constituent Tree Parsing"☆18Oct 17, 2020Updated 5 years ago
- Follow the Wisdom of the Crowd: Effective Text Generation via Minimum Bayes Risk Decoding☆20Nov 16, 2022Updated 3 years ago