Could you expand the capabilities of this library?

Good day!
Please tell me, could you work on this library?
I would like to try to train my Machine, but so far it seems not very convenient to me.
I would like to train the graph CoTM right away.
Can you help )))

Well, and besides, I would like you to make the library interface more convenient.

I will be so glad!
