Official implementation of AI-TOD-v2 and NWD-RKA - View it on GitHub