pytorch WME