A basic pure pytorch implementation of flash attention
☆17Oct 28, 2024Updated 2 years ago
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:
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆17Jun 18, 2024Updated 2 years ago
- Personal dotfiles.☆11May 29, 2026Updated 5 months ago
- ☆17Oct 22, 2024Updated 2 years ago
- Lessons for learning 3D computer vision in BoofCV☆14Apr 21, 2019Updated 7 years ago
- [CVPR '26] - PinPoint: Evaluation of Composed Image Retrieval with Explicit Negatives, Multi-Image Queries, and Paraphrase Testing☆15Jun 9, 2026Updated 4 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆20May 30, 2024Updated 2 years ago
- ☆95Jul 5, 2024Updated 2 years ago
- The code for the paper, 'Meta-Curvature, Eunbyung Park and Junier Oliver, NeurIPS 2019'☆11Jan 20, 2020Updated 6 years ago
- fuchsia: minimal libaray for RL on LLMs☆10Mar 8, 2026Updated 7 months ago
- ☆34May 14, 2025Updated last year
- Extensive time series analysis of chinese PM2.5 content, using models from ARMA and VAR to LSTMs and dynamic time warping clustering☆12Aug 17, 2019Updated 7 years ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆45Jan 15, 2024Updated 2 years ago
- Pyramid Mask Text Detector designed by SenseTime Video Intelligence Research team.☆14Aug 1, 2019Updated 7 years ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆62Jun 21, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Deep Learning for Computer Vision☆65Jun 19, 2024Updated 2 years ago
- ☆16Jul 1, 2026Updated 3 months ago
- 把 On-Policy Distillation 接到 DSpark。☆19Jun 28, 2026Updated 4 months ago
- Stable diffusion dedicated Hardware with multiple pipelined processor cores☆14Apr 9, 2026Updated 6 months ago
- A non-exhaustive collection of vision transformer models implemented in TensorFlow.☆10Sep 25, 2022Updated 4 years ago
- Optimized primitives for collective multi-GPU communication☆11May 8, 2024Updated 2 years ago
- example showing finetuning of fuyu☆10Jul 26, 2024Updated 2 years ago
- {DeepL, Google, WMT-Best, davinci-003, turbo, gpt-4} × {En-De, En-Cs, En-Ru, En-Zh, De-Fr, En-Ja, Uk-En, Uk-Cs, En-Hr, En-Ha, En-Is}☆14Jun 18, 2023Updated 3 years ago
- ☆16Apr 2, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- 1st place. Team Crane's solution the 9/6 Jane Street x GPU_MODE contest! @aswinkumar1999 @rathull @kyolebu☆12Oct 16, 2025Updated last year
- GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface☆20Sep 25, 2023Updated 3 years ago
- Reimplementation of Hierarchical Attentive Recurrent Tracking (Kosiorek et al., 2017) in PyTorch☆16Jul 29, 2018Updated 8 years ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆16May 10, 2024Updated 2 years ago
- Object recognition with Pepper using a deep learning model☆10Sep 16, 2021Updated 5 years ago
- ☆13Jul 22, 2026Updated 3 months ago
- Analyzing LLM Alignment via Token distribution shift☆17Jan 26, 2024Updated 2 years ago
- implementation of text to speech models☆13May 25, 2026Updated 5 months ago
- The best ChatGPT that $100 can buy.☆60Updated this week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Using FlexAttention to compute attention with different masking patterns☆47Sep 22, 2024Updated 2 years ago
- a bunch of rubrics I made in different format and structure for llm judge and other use cases☆16Sep 22, 2025Updated last year
- ☆52Jan 28, 2024Updated 2 years ago
- BLAS Library in Exo☆12Aug 15, 2026Updated 2 months ago
- ☆10Apr 24, 2021Updated 5 years ago
- Codes for the ICLR 2022 paper: Trigger Hunting with a Topological Prior for Trojan Detection☆11Sep 19, 2023Updated 3 years ago
- Seed repo for CVPR'21 Continual Learning Challenge☆13Apr 26, 2021Updated 5 years ago