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 10 months ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- ☆17Aug 1, 2025Updated 7 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
- This repository reproduces the results in the paper "How expressive are transformers in spectral domain for graphs?"(published in TMLR)☆12Jul 10, 2022Updated 3 years ago
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning☆21Nov 4, 2021Updated 4 years ago
- ☆11Dec 22, 2024Updated last year
- Spectral Graph Attention Network with Fast Eigen-approximation☆12Dec 24, 2021Updated 4 years ago
- A PyTorch implementation of the paper https://arxiv.org/abs/1709.04875☆10Jul 22, 2020Updated 5 years ago
- BFloat16 Fused Adam Operator for PyTorch☆16Nov 16, 2024Updated last year
- 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 last year
- Helpful tools and examples for working with flex-attention☆1,140Feb 8, 2026Updated 3 weeks ago
- ☆16Feb 24, 2026Updated last 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
- 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
- ☆36Feb 26, 2024Updated 2 years ago
- Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spaces, NeurIPS 2021☆14Dec 11, 2021Updated 4 years ago
- ☆124May 28, 2024Updated last year
- ☆31Jul 2, 2023Updated 2 years ago
- ☆32May 26, 2024Updated last year
- Codes for accepted paper "Cooperative Pruning in Cross-Domain Deep Neural Network Compression" in IJCAI 2019.☆12Aug 15, 2019Updated 6 years ago
- Pytorch (PyG) and Tensorflow (Keras/Spektral) implementation of Total Variation Graph Neural Network (TVGNN), as presented at ICML 2023.☆20Mar 15, 2025Updated 11 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆110Oct 11, 2025Updated 4 months ago
- ☆105Nov 7, 2024Updated last year
- Transformers components but in Triton☆34May 9, 2025Updated 9 months ago
- Triton implementation of FlashAttention2 that adds Custom Masks.☆168Aug 14, 2024Updated last year
- ☆16Oct 20, 2025Updated 4 months ago
- A basic pure pytorch implementation of flash attention☆16Oct 28, 2024Updated last year
- Memory-Based Meta-Learning on Non-Stationary Distributions☆17Mar 14, 2024Updated last year
- Efficient triton implementation of Native Sparse Attention.☆268May 23, 2025Updated 9 months ago
- STABILIZING GRADIENTS FOR DEEP NEURAL NETWORKS VIA EFFICIENT SVD PARAMETERIZATION☆16Jun 5, 2018Updated 7 years ago
- Simple python library for generating your own perfetto traces for your application. Can be used for both app instrumentation and custom …☆25Jun 22, 2025Updated 8 months ago
- [ICML 2024] Recurrent Distance Filtering for Graph Representation Learning☆15Jun 10, 2024Updated last year
- [BMVC 2022] Information Theoretic Representation Distillation☆19Oct 6, 2023Updated 2 years ago
- Triton kernels for Flux☆22Jul 7, 2025Updated 7 months ago
- A token pruning method that accelerates ViTs for various tasks while maintaining high performance.☆27Jul 21, 2025Updated 7 months ago
- Implementation of NAACL 2024 Outstanding Paper "LM-Infinite: Simple On-the-Fly Length Generalization for Large Language Models"☆151Mar 13, 2025Updated 11 months ago