XFormers: A collection of composable Transformer building blocks.
conda install selfexplainml::xformers