Comment by gbieler

This is very useful! Are you planning to support model training as well?