Fast and memory-efficient exact attention
☆20Jul 22, 2024Updated last year
Alternatives and similar repositories for flash-attention
Users that are interested in flash-attention are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A simple implementation of a deep linear Pytorch module☆21Oct 16, 2020Updated 5 years ago
- 因果学习、不确定性估计相关资源(论文、代码、数据、博客等)☆17Feb 23, 2024Updated 2 years ago
- A simple cycle-accurate DaDianNao simulator☆13Mar 27, 2019Updated 7 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆70Dec 24, 2022Updated 3 years ago
- Implementation of N-Grammer, augmenting Transformers with latent n-grams, in Pytorch☆76Dec 4, 2022Updated 3 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Official repository for "TrustGeoGen: Formal-Verified Data Engine for Trustworthy Multi-modal Geometric Problem Solving"☆23Sep 1, 2025Updated 8 months ago
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Mar 30, 2021Updated 5 years ago
- ☆15Jan 27, 2025Updated last year
- ☆13Apr 16, 2022Updated 4 years ago
- Block-Recurrent Dynamics in ViTs 🦖☆38Dec 24, 2025Updated 4 months ago
- [CVPR 2026 Fingdings] This repo is the official implementation of "Euclid’s Gift: Enhancing Spatial Perception and Reasoning in Vision‑La…☆28Mar 15, 2026Updated 2 months ago
- A simple and effective feature alignment method with proposed anchor loss for person re-identification☆15Aug 18, 2020Updated 5 years ago
- Hardware Division Units☆10Jul 17, 2014Updated 11 years ago
- FAQ for University of CaliforniaSanta Cruz 2019 Incoming Grads☆12Apr 4, 2019Updated 7 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- RADIX-4 SRT division☆12Oct 31, 2019Updated 6 years ago
- ☆10May 24, 2020Updated 5 years ago
- ChEMBL Database used to create Lipinski Descriptors (ADME Pharmokinetic Profile) to use in a Random Forest Regression Model☆15Nov 8, 2020Updated 5 years ago
- This project implements the Titans architecture from the paper "Titans: Learning to Memorize at Test Time" for market data prediction.☆11Jan 19, 2025Updated last year
- A Pytorch implementation of Global Self-Attention Network, a fully-attention backbone for vision tasks☆94Nov 21, 2020Updated 5 years ago
- Pixie pipeline described in Liu et al., Robust phenotyping of highly multiplexed tissue imaging data using pixel-level clustering☆12Aug 3, 2023Updated 2 years ago
- Pytorch implementation of "Very Deep Graph Neural Networks via Noise Regularisation"☆10Aug 22, 2021Updated 4 years ago
- ZET-Speech: Zero-shot adaptive Emotion-controllable Text-to-Speech Synthesis with Diffusion and Style-based Models (TTS)☆10Mar 9, 2024Updated 2 years ago
- Code for H. Narasimhan, "Learning with Complex Loss Functions and Constraints", AISTATS 2018☆11Mar 21, 2018Updated 8 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆15Apr 9, 2026Updated last month
- Implementation of "Denoise Pretraining on Non-equilibrium Molecular Conformations for Accurate and Transferable Neural Potentials" in PyT…☆14Jul 26, 2023Updated 2 years ago
- ☆16Oct 20, 2025Updated 6 months ago
- A simple cross attention that updates both the source and target in one step☆196Jul 29, 2025Updated 9 months ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Aug 30, 2022Updated 3 years ago
- Adaptive floating-point based numerical format for resilient deep learning☆14Apr 11, 2022Updated 4 years ago
- A PyTorch implementation of [VCT](https://github.com/google-research/google-research/tree/master/vct)☆10Nov 25, 2022Updated 3 years ago
- Neptune - TensorBoard integration 🧩 Experiment tracking with advanced UI, collaborative features, and user access management.☆13Sep 4, 2025Updated 8 months ago
- A implement of run-length encoding for Pytorch tensor using CUDA☆14Apr 7, 2021Updated 5 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Associative scan package for DRYing some code between repos☆18Jan 5, 2026Updated 4 months ago
- 9th solution☆11Oct 11, 2022Updated 3 years ago
- Showing full TensorBoard support in Tensorflow for a CNN using MNIST data.☆13Oct 19, 2019Updated 6 years ago
- Text to Speech with PyTorch (English and Mongolian)☆13May 3, 2020Updated 6 years ago
- unsigned Radix-2 SRT division,基2除法☆16May 12, 2015Updated 11 years ago
- ☆12Jun 12, 2017Updated 8 years ago
- ☆10Jan 29, 2021Updated 5 years ago