About Code release for "FlashBias: Fast Computation of Attention with Bias" (NeurIPS 2025), https://arxiv.org/abs/2505.12044
☆27Nov 17, 2025Updated 5 months ago
Alternatives and similar repositories for FlashBias
Users that are interested in FlashBias are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Generalizing to New Physical Systems via Context-Informed Dynamics Model☆28May 6, 2024Updated last year
- ☆25May 21, 2024Updated last year
- a mini TPU with floating point arithmetic☆53Dec 22, 2025Updated 3 months ago
- About Code release for “RoPINN: Region Optimized Physics-Informed Neural Networks” (NeurIPS 2024), https://arxiv.org/abs/2405.14369☆60Jul 14, 2025Updated 9 months ago
- A library to perform targeted free energy perturbation with normalizing flows.☆11Sep 1, 2025Updated 7 months ago
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- official repo for `thinking with images through-self-calling`☆26Dec 28, 2025Updated 3 months ago
- ☆29Jul 8, 2024Updated last year
- Group project "Algorithms for large-scale optimal transport". Implement ADMMs and Sinkhorn's Algorithms.☆11Jan 28, 2019Updated 7 years ago
- implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)☆17Feb 24, 2022Updated 4 years ago
- Distributed SDDMM Kernel☆12Jul 8, 2022Updated 3 years ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆11Nov 5, 2024Updated last year
- PyTorch implementation for "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆13Jul 21, 2024Updated last year
- [ACL 2021 Findings] HySPA: Hybrid Span Generation for Scalable Text-to-Graph Extraction☆10Sep 16, 2021Updated 4 years ago
- Webserver & command-line tool for search of APO (unbound) protein structures from HOLO (bound) forms and vice versa. http://apoholo.cz☆19Jun 14, 2024Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆15Jan 26, 2023Updated 3 years ago
- [EMNLP 2022] Language Model Pre-Training with Sparse Latent Typing☆14Feb 10, 2023Updated 3 years ago
- ☆13Sep 13, 2023Updated 2 years ago
- Official repo for BWLer: Barycentric Weight Layer☆30Mar 20, 2026Updated 3 weeks ago
- GEMV implementation with CUTLASS☆21Aug 21, 2025Updated 7 months ago
- ☆19Mar 5, 2024Updated 2 years ago
- Simple differentiable approximate ocean models built with JAX.☆17Oct 13, 2023Updated 2 years ago
- All-atom inverse protein folding through discrete flow matching (ICML2025)☆25Mar 22, 2026Updated 3 weeks ago
- ☆22Dec 19, 2025Updated 4 months 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.
- ☆18Apr 19, 2024Updated 2 years ago
- ☆57Apr 1, 2025Updated last year
- Latent Diffusion model for ensemble weather forecasting☆24Mar 4, 2026Updated last month
- Aligned Diffusion Schroedinger Bridges (UAI 2023)☆14Sep 18, 2025Updated 7 months ago
- Study materials about "Deep Learning for Molecular Applications".☆15Aug 5, 2019Updated 6 years ago
- ☆13Dec 19, 2023Updated 2 years ago
- ☆10Feb 21, 2023Updated 3 years ago
- Sparse Backpropagation for Mixture-of-Expert Training☆30Jul 2, 2024Updated last year
- High-performance LLM operator library built on TileLang.☆102Updated this week
- 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.
- python code for Multi-Coordinate Driving (MCD) method☆15Jul 16, 2024Updated last year
- Code publication to the paper "Normalized Attention Without Probability Cage"☆17Nov 9, 2021Updated 4 years ago
- ☆22Sep 1, 2025Updated 7 months ago
- A distance distribution-based visualization method for atomic pair energy and precise prediction of absolute binding free energy.☆17Feb 4, 2025Updated last year
- Highly accurate and efficient deep learning paradigm for full-atom protein loop modeling with KarmaLoop☆15Jun 9, 2024Updated last year
- Official Code Repository for the paper "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆16Jul 21, 2024Updated last year
- EquiformerV3: Scaling Efficient, Expressive, and General SE(3)-Equivariant Graph Attention Transformers☆59Apr 13, 2026Updated last week