TransformerX
TransformerX copied to clipboard
MultiheadAttention: Add support for distributed computation
Add the capability to support distributed computation