A basic pure pytorch implementation of flash attention
☆16Oct 28, 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:
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Jun 18, 2024Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆16Jun 16, 2024Updated last year
- Official Documentation for DSPy Library☆23Mar 27, 2026Updated 2 weeks ago
- ☆29Jul 8, 2024Updated last year
- notebooks to finetune `bert-small-amharic`, `bert-mini-amharic`, and `xlm-roberta-base` models using an Amharic text classification datas…☆11May 10, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Muday Expense Tracking App☆14Jan 5, 2025Updated last year
- Implementation of "Denoise Pretraining on Non-equilibrium Molecular Conformations for Accurate and Transferable Neural Potentials" in PyT…☆14Jul 26, 2023Updated 2 years ago
- Experimental GPU language with meta-programming☆27Sep 6, 2024Updated last year
- Official implementation of Scalable Transformer for PDE surrogate modelling☆55Mar 23, 2024Updated 2 years ago
- ☆20May 30, 2024Updated last year
- [SIGGRAPH Asia 2025] "ASIA: Adaptive 3D Segmentation using Few Image Annotations ".☆26Feb 14, 2026Updated last month
- FlexiTokens☆19Dec 27, 2025Updated 3 months ago
- A repository with some Deep Reinforcement Learning baselines written in julia using Flux.☆12Mar 28, 2023Updated 3 years ago
- Nomalizing flows for orbita-free DFT☆11Sep 20, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- [ICLR 2025] Neural Operator-Assisted Computational Fluid Dynamics in PyTorch☆80Nov 21, 2025Updated 4 months ago
- ☆26Jan 26, 2026Updated 2 months ago
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Jun 21, 2023Updated 2 years ago
- ☆34May 14, 2025Updated 10 months ago
- Official implementation for RoMaP :Robust 3D-Masked Part-level Editing in 3D Gaussian Splatting with Regularized Score Distillation Sampl…☆20Aug 5, 2025Updated 8 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆43Jan 15, 2024Updated 2 years ago
- Deep Learning for Computer Vision☆61Jun 19, 2024Updated last year
- ☆13May 7, 2023Updated 2 years ago
- ☆13Dec 27, 2020Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- Extensive time series analysis of chinese PM2.5 content, using models from ARMA and VAR to LSTMs and dynamic time warping clustering☆11Aug 17, 2019Updated 6 years ago
- MLX Swift implementation of Andrej Karpathy's Let's build GPT video☆63Apr 14, 2024Updated last year
- ☆13Aug 22, 2024Updated last year
- ☆15Oct 24, 2023Updated 2 years ago
- Stable diffusion dedicated Hardware with multiple pipelined processor cores☆14Mar 5, 2026Updated last month
- Bilinear interpolation on grids with jax☆16May 13, 2022Updated 3 years ago
- Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆15May 10, 2024Updated last year
- A tutorial on doing RL research in Julia using both Jupyter notebooks and normal project structures.☆10Jun 23, 2021Updated 4 years ago
- ☆39Mar 3, 2026Updated last month
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- CUDA implementation of Wavelet KAN.☆17Jun 8, 2024Updated last year
- Experiment utility code, specifically designed for use with Compute Canada.☆11Jan 27, 2025Updated last year
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆606Oct 7, 2025Updated 6 months ago
- [ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data☆65Mar 8, 2026Updated last month
- ☆51Jan 28, 2024Updated 2 years ago
- GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface☆19Sep 25, 2023Updated 2 years ago
- Rheological Universal Differential Equations: scientific machine learning for modeling complex fluids☆21Dec 23, 2022Updated 3 years ago