В pymorphy не стали добавлять связи между типами в *отдельном API, потому что сразу возникают вопросы, какие связи нужны, а какие нет (брань-бранное-отобранное-отобранность-отобрано-отобранный-отбираемый-отбирать-брать-беру-берущий-берущийся , а теперь назовите каждую пару отношений в обе стороны, пожалуйста, по частям речи тут на каждую по несколько форм получается), в каком виде их представлять, как компактно хранить (было актуально в 2012м!), да и датасетов мало было в 2012-2015м. Но в opencorpora, откуда с 2015г берётся БД для pymorphy, часть отношений между словами есть (помогайте контрибутить остальные!).
Берите дамп opencorpora для калибровки, и вперёд писать универсальную и качественную ML модель для этой задачи (DL плох, когда мало примеров каждого класса и много исключений, поэтому морфология и семантика русского плохо ему даётся, ну вы понимаете).