A basic pure pytorch implementation of flash attention
☆17Oct 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:
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆16Jun 16, 2024Updated 2 years ago
- Here you find the Spectral Operator Learning Under construcTION☆20May 8, 2024Updated 2 years ago
- ☆29Jul 8, 2024Updated last year
- ☆17Oct 22, 2024Updated 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
- 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.
- Official implementation of Scalable Transformer for PDE surrogate modelling☆57Mar 23, 2024Updated 2 years ago
- ☆20May 30, 2024Updated 2 years ago
- [SIGGRAPH Asia 2025] "ASIA: Adaptive 3D Segmentation using Few Image Annotations ".☆26Feb 14, 2026Updated 4 months ago
- Source code for paper "Learning the Solution Operator of Boundary Value Problems using Graph Neural Networks"☆20Jul 25, 2024Updated last year
- 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☆12Sep 20, 2024Updated last year
- Demonstration that finetuning RoPE model on larger sequences than the pre-trained model adapts the model context limit☆63Jun 21, 2023Updated 3 years ago
- ☆29Jan 26, 2026Updated 5 months ago
- ☆34May 14, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆10Nov 15, 2023Updated 2 years ago
- Official implementation for RoMaP :Robust 3D-Masked Part-level Editing in 3D Gaussian Splatting with Regularized Score Distillation Sampl…☆22Aug 5, 2025Updated 10 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆43Jan 15, 2024Updated 2 years ago
- ☆13May 7, 2023Updated 3 years ago
- FlexiTokens☆23Dec 27, 2025Updated 6 months ago
- [ICML 2024] Official PyTorch implementation of the Vectorized Conditional Neural Field.☆18Aug 1, 2024Updated last year
- ☆13Dec 27, 2020Updated 5 years ago
- 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 6 years ago
- An API designed for code completion and fine-tuning of open-source large language models on internal codebases and documents.☆14Oct 17, 2024Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆17May 24, 2023Updated 3 years ago
- Official implementation of the paper "Pretraining Language Models to Ponder in Continuous Space"☆26Jul 21, 2025Updated 11 months 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
- Using FlexAttention to compute attention with different masking patterns☆47Sep 22, 2024Updated last year
- Webcam demo for SKTBrain/DiscoGAN☆12Sep 11, 2019Updated 6 years ago
- Bilinear interpolation on grids with jax☆16May 13, 2022Updated 4 years ago
- Multi-lingual large voice generation model, providing inference, training and deployment full-stack ability.☆13Jul 15, 2024Updated last year
- A tutorial on doing RL research in Julia using both Jupyter notebooks and normal project structures.☆10Jun 23, 2021Updated 5 years ago
- Continual Learning Toolkit for Reinforcement Learning☆21Jan 28, 2018Updated 8 years ago
- 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.
- CUDA implementation of Wavelet KAN.☆17Jun 8, 2024Updated 2 years ago
- ☆40May 8, 2026Updated last month
- Single File, Single GPU, From Scratch, Efficient, Full Parameter Tuning library for "RL for LLMs"☆625Oct 7, 2025Updated 8 months ago
- [NeurIPS 2024 Spotlight] Towards Universal Mesh Movement Networks☆19Jul 16, 2025Updated 11 months ago
- ☆52Jan 28, 2024Updated 2 years ago
- GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface☆20Sep 25, 2023Updated 2 years ago
- Rheological Universal Differential Equations: scientific machine learning for modeling complex fluids☆21Dec 23, 2022Updated 3 years ago