that can be definitely implemented, I've also similar idea in another repo in my GitHub called ckpt you can check if you are interested but definitely that's sth that can be added ofc