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:
- 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
- ☆11Dec 22, 2024Updated last year
- Spectral Graph Attention Network with Fast Eigen-approximation☆12Dec 24, 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
- Repository of the paper "On the Trade-off between Over-smoothing and Over-squashing in Deep Graph Neural Networks" published in ACM CIKM …☆18Aug 8, 2023Updated 2 years ago
- ☆15Oct 30, 2025Updated 3 months ago
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- Expressive diffeomorphic transformations based on the closed-form integration of continuous piecewise affine velocity functions.☆16Aug 9, 2023Updated 2 years ago
- ☆124May 28, 2024Updated last year
- ☆31Jul 2, 2023Updated 2 years ago
- ☆32May 26, 2024Updated last year
- 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
- Codes for accepted paper "Cooperative Pruning in Cross-Domain Deep Neural Network Compression" in IJCAI 2019.☆12Aug 15, 2019Updated 6 years ago
- ☆104Nov 7, 2024Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆109Oct 11, 2025Updated 4 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆167Aug 14, 2024Updated last year
- Memory-Based Meta-Learning on Non-Stationary Distributions☆17Mar 14, 2024Updated last year
- Efficient triton implementation of Native Sparse Attention.☆262May 23, 2025Updated 8 months 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
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learning☆15Jun 10, 2024Updated last year
- Triton kernels for Flux☆22Jul 7, 2025Updated 7 months ago
- A token pruning method that accelerates ViTs for various tasks while maintaining high performance.☆28Jul 21, 2025Updated 6 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
- [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
- Code for “ACE-HGNN: Adaptive Curvature ExplorationHyperbolic Graph Neural Network”☆17Mar 3, 2022Updated 3 years ago
- Efficient PScan implementation in PyTorch☆17Jan 2, 2024Updated 2 years ago
- ☆15Dec 5, 2019Updated 6 years ago
- ☆18Dec 2, 2024Updated last year
- ☆18Dec 9, 2023Updated 2 years ago
- [NAACL 2025] MiLoRA: Harnessing Minor Singular Components for Parameter-Efficient LLM Finetuning☆19May 31, 2025Updated 8 months ago