Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

jinaai
/
jina-bert-flash-implementation

Transformers
bert
custom_code
๐Ÿ‡ช๐Ÿ‡บ Region: EU
Model card Files Files and versions
xet
Community
18
jina-bert-flash-implementation / flash_components
74.2 kB
  • 6 contributors
History: 1 commit
Markus28's picture
Markus28
feat: moved flash attention code into this repository
46df05d almost 2 years ago
  • bert_padding.py
    9.63 kB
    feat: moved flash attention code into this repository almost 2 years ago
  • block.py
    17.3 kB
    feat: moved flash attention code into this repository almost 2 years ago
  • embedding.py
    6.27 kB
    feat: moved flash attention code into this repository almost 2 years ago
  • mha.py
    35 kB
    feat: moved flash attention code into this repository almost 2 years ago
  • mlp.py
    6.03 kB
    feat: moved flash attention code into this repository almost 2 years ago