Skip to main content

Triton implemention of the Energy Based Transformer attention mechanism

Project description

EBT Flash Attention

Triton implemention of the Energy Based Transformer attention mechanism

Description

Leverage Flash Attention Triton kernel i

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ebt_flash_attention-0.1.0.tar.gz (1.4 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

ebt_flash_attention-0.1.0-py3-none-any.whl (18.7 kB view details)

Uploaded Python 3

File details

Details for the file ebt_flash_attention-0.1.0.tar.gz.

File metadata

  • Download URL: ebt_flash_attention-0.1.0.tar.gz
  • Upload date:
  • Size: 1.4 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.8.0

File hashes

Hashes for ebt_flash_attention-0.1.0.tar.gz
Algorithm Hash digest
SHA256 83d04f439e707d72b8b4bddbc17a83583fe65af06b8d3d2b6fd11a06f337c723
MD5 f9405448e721515586343952576ac90c
BLAKE2b-256 d3f1635cf4fc85e0ffa010988f1b09de176a68ef090051423e40d4939ae3916d

See more details on using hashes here.

File details

Details for the file ebt_flash_attention-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for ebt_flash_attention-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b57e345291d8fce626103b97c888cef5903d189aadad27d524e7d51c620955e2
MD5 0ddc81ffd41ad0720c411e300e16da23
BLAKE2b-256 e77d808e3b369d27fe250f23fff38ba4bd0b1af946672e008045d6cbbbfb0678

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page