* chunked cross entropy loss * refactor so we can add test * use relative import * update schema description
Wing Lian