There are 2 repositories under large-output-space topic.
Official codebase for NeurIPS 2022 paper End-to-end Learning to Index and Search in Large Output Spaces
Code for DEXA