xformers 0.0.34.dev1102__tar.gz → 0.0.34.dev1116__tar.gz

xformers-0.0.34.dev1116/MANIFEST.in

@@ -0,0 +1,17 @@
+include LICENSE
+include requirements.txt
+include version.txt
+
+recursive-include xformers/csrc *
+recursive-include third_party/cutlass/include *
+recursive-include third_party/cutlass/tools/util/include *
+recursive-include third_party/cutlass/examples *
+recursive-include third_party/flash-attention/csrc *
+recursive-include third_party/flash-attention/flash_attn *
+recursive-include third_party/flash-attention/hopper *
+
+prune third_party/flash-attention/csrc/cutlass/docs/
+prune third_party/flash-attention/csrc/cutlass/test/
+prune third_party/flash-attention/csrc/cutlass/tools/
+prune third_party/flash-attention/csrc/cutlass/media/
+prune third_party/flash-attention/csrc/cutlass/python/

xformers-0.0.34.dev1116/PKG-INFO

@@ -0,0 +1,31 @@
+Metadata-Version: 2.4
+Name: xformers
+Version: 0.0.34.dev1116
+Summary: XFormers: A collection of composable Transformer building blocks.
+Home-page: https://facebookresearch.github.io/xformers/
+Author: Facebook AI Research
+Author-email: oncall+xformers@xmail.facebook.com
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.10
+Requires-Dist: numpy
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+XFormers: A collection of composable Transformer building blocks.XFormers aims at being able to reproduce most architectures in the Transformer-family SOTA,defined as compatible and combined building blocks as opposed to monolithic models

xformers-0.0.34.dev1116/README.md

@@ -0,0 +1,135 @@
+<img src="./docs/assets/logo.png" width=800>
+
+[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/facebookresearch/xformers/blob/main/docs/source/xformers_mingpt.ipynb)
+<br/><!--
+![PyPI](https://img.shields.io/pypi/v/xformers)
+![PyPI - License](https://img.shields.io/pypi/l/xformers)
+[![Documentation Status](https://github.com/facebookresearch/xformers/actions/workflows/gh-pages.yml/badge.svg)](https://github.com/facebookresearch/xformers/actions/workflows/gh-pages.yml/badge.svg)
+-->
+[![CircleCI](https://circleci.com/gh/facebookresearch/xformers.svg?style=shield)](https://app.circleci.com/pipelines/github/facebookresearch/xformers/)
+[![Codecov](https://codecov.io/gh/facebookresearch/xformers/branch/main/graph/badge.svg?token=PKGKDR4JQM)](https://codecov.io/gh/facebookresearch/xformers)
+[![black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+<br/>
+[![PRs welcome](https://img.shields.io/badge/PRs-welcome-brightgreen.svg)](CONTRIBUTING.md)
+<!--
+[![Downloads](https://pepy.tech/badge/xformers)](https://pepy.tech/project/xformers)
+-->
+--------------------------------------------------------------------------------
+
+## xFormers - Toolbox to Accelerate Research on Transformers
+
+xFormers is:
+- **Customizable building blocks**: Independent/customizable building blocks that can be used without boilerplate code. The components are domain-agnostic and xFormers is used by researchers in vision, NLP and more.
+- **Research first**: xFormers contains bleeding-edge components, that are not yet available in mainstream libraries like PyTorch.
+- **Built with efficiency in mind**: Because speed of iteration matters, components are as fast and memory-efficient as possible. xFormers contains its own CUDA kernels, but dispatches to other libraries when relevant.
+
+## Installing xFormers
+
+* **(RECOMMENDED, linux & win) Install latest stable with pip**: Requires [PyTorch 2.10.0](https://pytorch.org/get-started/locally/)
+
+```bash
+# [linux & win] cuda 12.6 version
+pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu126
+# [linux & win] cuda 12.8 version
+pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu128
+# [linux & win] cuda 13.0 version
+pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu130
+# [linux only] (EXPERIMENTAL) rocm 7.1 version
+pip3 install -U xformers --index-url https://download.pytorch.org/whl/rocm7.1
+```
+
+* **Development binaries**:
+
+```bash
+# Same requirements as for the stable version above
+pip install --pre -U xformers
+```
+
+* **Install from source**: If you want to use with another version of PyTorch for instance (including nightly-releases)
+
+```bash
+# (Optional) Makes the build much faster
+pip install ninja
+# Set TORCH_CUDA_ARCH_LIST if running and building on different GPU types
+# NOTE: pytorch must already be installed!
+pip install -v --no-build-isolation -U git+https://github.com/facebookresearch/xformers.git@main#egg=xformers
+# (this can take dozens of minutes)
+```
+
+
+## Benchmarks
+
+**Memory-efficient MHA**
+![Benchmarks for ViTS](./docs/plots/mha/mha_vit.png)
+*Setup: A100 on f16, measured total time for a forward+backward pass*
+
+Note that this is exact attention, not an approximation, just by calling [`xformers.ops.memory_efficient_attention`](https://facebookresearch.github.io/xformers/components/ops.html#xformers.ops.memory_efficient_attention)
+
+**More benchmarks**
+
+xFormers provides many components, and more benchmarks are available in [BENCHMARKS.md](BENCHMARKS.md).
+
+### (Optional) Testing the installation
+
+This command will provide information on an xFormers installation, and what kernels are built/available:
+
+```python
+python -m xformers.info
+```
+
+## Using xFormers
+
+### Key Features
+
+1. Optimized building blocks, beyond PyTorch primitives
+   1. Memory-efficient exact attention - up to 10x faster
+   2. sparse attention
+   3. block-sparse attention
+   4. fused softmax
+   5. fused linear layer
+   6. fused layer norm
+   7. fused dropout(activation(x+bias))
+   8. fused SwiGLU
+
+### Install troubleshooting
+
+
+* NVCC and the current CUDA runtime match. Depending on your setup, you may be able to change the CUDA runtime with `module unload cuda; module load cuda/xx.x`, possibly also `nvcc`
+* the version of GCC that you're using matches the current NVCC capabilities
+* the `TORCH_CUDA_ARCH_LIST` env variable is set to the architectures that you want to support. A suggested setup (slow to build but comprehensive) is `export TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6"`
+* If the build from source OOMs, it's possible to reduce the parallelism of ninja with `MAX_JOBS` (eg `MAX_JOBS=2`)
+* If getting error message `Filename longer than 260 characters` on Windows, make sure long paths are enabled at OS level, and also execute the command `git config --global core.longpaths true`
+
+
+### License
+
+xFormers has a BSD-style license, as found in the [LICENSE](LICENSE) file.
+It includes code from the [triton-lang/kernels](https://github.com/triton-lang/kernels) repo.
+
+## Citing xFormers
+
+If you use xFormers in your publication, please cite it by using the following BibTeX entry.
+
+``` bibtex
+@Misc{xFormers2022,
+  author =       {Benjamin Lefaudeux and Francisco Massa and Diana Liskovich and Wenhan Xiong and Vittorio Caggiano and Sean Naren and Min Xu and Jieru Hu and Marta Tintore and Susan Zhang and Patrick Labatut and Daniel Haziza and Luca Wehrstedt and Jeremy Reizenstein and Grigory Sizov},
+  title =        {xFormers: A modular and hackable Transformer modelling library},
+  howpublished = {\url{https://github.com/facebookresearch/xformers}},
+  year =         {2022}
+}
+```
+
+## Credits
+
+The following repositories are used in xFormers, either in close to original form or as an inspiration:
+
+* [Sputnik](https://github.com/google-research/sputnik)
+* [GE-SpMM](https://github.com/hgyhungry/ge-spmm)
+* [Triton](https://github.com/openai/triton)
+* [LucidRain Reformer](https://github.com/lucidrains/reformer-pytorch)
+* [RevTorch](https://github.com/RobinBruegger/RevTorch)
+* [Nystromformer](https://github.com/mlpen/Nystromformer)
+* [FairScale](https://github.com/facebookresearch/fairscale/)
+* [Pytorch Image Models](https://github.com/rwightman/pytorch-image-models)
+* [CUTLASS](https://github.com/nvidia/cutlass)
+* [Flash-Attention](https://github.com/HazyResearch/flash-attention)

xformers-0.0.34.dev1116/pyproject.toml

@@ -0,0 +1,11 @@
+[build-system]
+# XXX: If your project needs other packages to build properly, add them to this list.
+requires = ["setuptools >= 64", "torch >= 2.10"]
+build-backend = "setuptools.build_meta"
+
+[tool.black]
+target-version = ["py310"]
+line-length = 88
+
+[tool.usort]
+first_party_detection = false

xformers-0.0.34.dev1116/requirements.txt

@@ -0,0 +1,4 @@
+# Example requirement, can be anything that pip knows
+# install with `pip install -r requirements.txt`, and make sure that CI does the same
+torch >= 2.10
+numpy