jinaai / jina-bert-flash-implementation

If you would like to use fused MLPs (e.g. to use activation checkpointing), you may install fused-dense also from source: console cd csrc/fuseddenselib && python setup.py install