About Code release for "FlashBias: Fast Computation of Attention with Bias" (NeurIPS 2025), https://arxiv.org/abs/2505.12044
☆33Nov 17, 2025Updated 9 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:
- About Code release for “RoPINN: Region Optimized Physics-Informed Neural Networks” (NeurIPS 2024), https://arxiv.org/abs/2405.14369☆65Jul 14, 2025Updated last year
- ☆26Sep 1, 2025Updated 11 months ago
- The Polaris datasets and benchmarks recipes☆15May 26, 2025Updated last year
- ☆14Aug 22, 2024Updated last year
- Code for ICML2025 Paper "Physics-informed Temporal Alignment for Auto-regressive PDE Foundation Models"☆17Jun 27, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆15May 9, 2024Updated 2 years ago
- [AAAI 2025] AeroGTO: An Efficient Graph-Transformer Operator for Learning Large-Scale Aerodynamics of 3D Vehicle Geometries☆23Jul 20, 2025Updated last year
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆30Jan 22, 2026Updated 6 months ago
- Generalizing to New Physical Systems via Context-Informed Dynamics Model☆28May 6, 2024Updated 2 years ago
- ☆94Jul 18, 2023Updated 3 years ago
- [NeurIPS 2025] Official PyTorch implementation of the CALM layers and CALM-PDE model.☆21Oct 20, 2025Updated 9 months ago
- Official Code Repository for the paper "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆16Jul 21, 2024Updated 2 years ago
- [TMLR 2025] Mesh-Informed Neural Operator : A Transformer Generative Approach☆19Jun 1, 2026Updated 2 months ago
- All-atom inverse protein folding through discrete flow matching (ICML2025)☆43Mar 22, 2026Updated 4 months 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.
- NeurIPS-2025☆21Nov 4, 2025Updated 9 months ago
- [Communications AI & Computing 2026]An Efficient Graph-Transformer Operator for Learning Physical Dynamics with Manifolds Embedding☆22Jun 24, 2026Updated last month
- ☆80May 27, 2026Updated 2 months ago
- About code release of "Transolver: A Fast Transformer Solver for PDEs on General Geometries", ICML 2024 Spotlight. https://arxiv.org/abs/…☆400Feb 26, 2026Updated 5 months ago
- Triton Implementation of Flash Attention with Bias.☆25Apr 16, 2025Updated last year
- ☆19Mar 25, 2025Updated last year
- ☆26May 21, 2024Updated 2 years ago
- Platonic Transformers: A Solid Choice For Equivariance (ICML 2026)☆63Jun 4, 2026Updated 2 months ago
- A benchmark for comprehensive evaluation on protein structure tokenization methods☆56Aug 2, 2025Updated last year
- 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.
- PDE-Transformer is a neural network architecture designed to efficiently process and predict the evolution of physical systems described …☆88Apr 11, 2026Updated 4 months ago
- [NeurIPS 2025 Spotlight] E2Former: An Efficient and Equivariant Transformer with Linear-Scaling Tensor Products☆30Feb 16, 2026Updated 6 months ago
- A Library for Advanced Neural PDE Solvers.☆335Mar 10, 2026Updated 5 months ago
- Audio Entailment: Deductive Reasoning for Audio Understanding☆17Dec 10, 2024Updated last year
- My list of essential Linux applications.☆30Mar 13, 2026Updated 5 months ago
- ☆29Jun 12, 2025Updated last year
- JAX translation of boltz☆31Aug 11, 2026Updated last week
- Community-maintained Boltz fork with bug fixes, broader compatibility, and CI.☆62Aug 7, 2026Updated last week
- Calculation of permutation-invariant root-mean-square-deviation of atomic coordinates☆18Jun 17, 2026Updated 2 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.
- ☆29Jul 8, 2024Updated 2 years ago
- An implementation of ESM2 in Equinox+JAX☆36Apr 20, 2026Updated 3 months ago
- Code to generate figures of paper "When do spectral gradient updates help in deep learning?"☆16Dec 3, 2025Updated 8 months ago
- Implementation for "DeltaPhi: Learning Physical Trajectory Residual for PDE Solving"☆13Jun 17, 2024Updated 2 years ago
- Distributed SDDMM Kernel☆13Jul 8, 2022Updated 4 years ago
- Code for Mesh Transformer describes in the EAGLE dataset☆42Feb 21, 2025Updated last year
- Muon in Int8 Precision Made Possible☆20Jun 18, 2026Updated 2 months ago