ok before one of you tries to assassinate me for building AGI, i figured out the bug. at least it's an interesting one so we're doing contrastive learning, which is a matching task between (query, document) pairs query 1 matches to document 1, query 2 matches to document