Is there still no memory and time-efficient way to do broadcasted matrix multiplication?  It seems like this would be essential for something like a transformer network since otherwise dot-product attention would require you to copy both your keys and your queries?  Is there a special trick to do dot-product attention in particular?