Official code for "Interaction Modeling with Multiplex Attention" (Neurips 2022) - View it on GitHub