would be nice to have it cross platform and cross library (for tensorflow, pytorch ... :)