Feature/tipi checkpoint