xformers

Форк

JFR
Jeremy Francis Reizenstein
memory_efficient_attention_partial
год назад
fe0526b

Описание

Hackable and optimized Transformers building blocks, supporting a composable construction.

Избранное0

Следят1

Языки

Python

C++
Cuda
C
Shell

[refactor] Update checkpoint to call into torch.utils.checkpoint (fairinternal/xformers#954)

год назад

.github

update flash-attention to v2.5.2 (fairinternal/xformers#995)

год назад

docs

[fmha] Added `LocalAttentionFromBottomRightMask`

2 года назад

examples

whitespace fixes

год назад

packaging

[ci] fix compute_wheel_version.py

2 года назад

stubs

[CI] Bumping up black (#315)

3 года назад

tests

memory_efficient_attention_partial

год назад

third_party

update flash-attention to v2.5.2 (fairinternal/xformers#995)

год назад

xformers

memory_efficient_attention_partial

год назад

.clang-format

Initial commit

4 года назад

.coveragerc

Refactor3: Embed flash's PY files as xformers._flash_attn

3 года назад

.editorconfig

Initial commit

4 года назад

.flake8

Revamp CI (mostly conda builds)

2 года назад

.gitignore

[feat] cutlass FlashAttention bias+dropout support (#587)

2 года назад

.gitmodules

Update FlashAttention v2

2 года назад

.isort.cfg

Refactor3: Embed flash's PY files as xformers._flash_attn

3 года назад

.markdownlint.json

Initial commit

4 года назад

.pre-commit-config.yaml

Update flake8 version in CI from 3.8.4 (May 2020) to 6.1.0 (July 2023)

2 года назад

.pyre_configuration

Refactor3: Embed flash's PY files as xformers._flash_attn

3 года назад

BENCHMARKS.md

Fix some internal lints

2 года назад

CHANGELOG.md

Export recent MHA changes from fbcode

год назад

4 года назад

год назад

Remove NVFuser from xFormers (#843)

2 года назад

LICENSE

Adding the license title (#507)

3 года назад

MANIFEST.in

[sdist] Fix detection of flash-attention verison. Remove some junk from cutlass as well

2 года назад

README.md

Add Luca, Jeremy and Grigory as a contributors (#986)

год назад

codecov.yml

Remove experimental folder (#851)

2 года назад

requirements-benchmark.txt

(empty commit message)

2 года назад

requirements-lra.txt

[feat] Compositional attention (#178)

3 года назад

requirements-test.txt

Don't install Triton nightly separately in CI

год назад

requirements.txt

[refactor] Update checkpoint to call into torch.utils.checkpoint (fairinternal/xformers#954)

год назад

setup.cfg

Initial commit

4 года назад

setup.py

update flash-attention to v2.5.2 (fairinternal/xformers#995)

год назад

version.txt

Bumpp dev version to 0.0.25 (fairinternal/xformers#1024)

год назад

Readme

README.md

Описание

Hackable and optimized Transformers building blocks, supporting a composable construction.

Избранное0

Следят1

Языки

Python

C++
Cuda
C
Shell

Сообщить о нарушении

Описание

Hackable and optimized Transformers building blocks, supporting a composable construction.

Избранное0

Следят1

Языки

Python

C++
Cuda
C
Shell

Сообщить о нарушении

xformers

JFRJeremy Francis Reizensteinmemory_efficient_attention_partialгод назадfe0526b

Описание

Языки

Описание

Языки

Описание

Языки

JFR
Jeremy Francis Reizenstein
memory_efficient_attention_partial
год назад
fe0526b