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

  • 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 over 1 year ago
  • bert_padding.py
    9.63 kB
    feat: moved flash attention code into this repository over 1 year ago
  • block.py
    17.3 kB
    feat: moved flash attention code into this repository over 1 year ago
  • embedding.py
    6.27 kB
    feat: moved flash attention code into this repository over 1 year ago
  • mha.py
    35 kB
    feat: moved flash attention code into this repository over 1 year ago
  • mlp.py
    6.03 kB
    feat: moved flash attention code into this repository over 1 year ago