Code

Adding GPU paged attention kernel#25839

Code

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom Rifur13:pagedRifur13/jax:pagedCopy head branch name to clipboardJan 24, 2025

+595

Commits

Commits on Jan 24, 2025

• Adding GPU paged attention kernel

  Rifur13

  committed
  e0b38f4

  Copy full SHA for e0b38f4