xFormers

Other Free

Optimized transformer building blocks for efficient training/inference.

About xFormers

xFormers provides fast attention implementations for PyTorch.