Code for NeurIPS 2025 paper "Distance-informed Neural Processes" - View it on GitHub